Commit 76a2ee3b authored by Anthony Larcher's avatar Anthony Larcher
Browse files

add loggings

parent 4b07087f
......@@ -1207,11 +1207,9 @@ def run_active_learning_tree(link,
def allies_within_show_hal(model_cfg,
model,
show,
dir_output_system,
init_diar,
vec_per_cluster,
vec_per_seg,
tmp_path,
file_info,
uem,
ref,
......@@ -1236,6 +1234,12 @@ def allies_within_show_hal(model_cfg,
"""
der_track_show = {}
hal_dir = f"{model_cfg['tmp_dir']}/seg/first_th{model_cfg['first_seg']['thr_h']}" \
f"/second_th{model_cfg['within_show']['th_w']}/hal/"
if not os.path.isdir(hal_dir):
os.makedirs(hal_dir)
#####################################################################################################
# Creat dendrogram by using upper level of current clustering and segments in each cluster######
#####################################################################################################
......@@ -1248,7 +1252,7 @@ def allies_within_show_hal(model_cfg,
model_cfg,
show,
init_diar,
tmp_path)
hal_dir)
bottomline_cluster_list = bottomline_diar.unique('cluster')
......
......@@ -36,7 +36,6 @@ import pandas
import pickle
import scipy
import sidekit
import sklearn
import s4d
import torch
import yaml
......@@ -711,7 +710,7 @@ def lium_xv_initial_training(model_cfg, train_data):
model_iv.plda_f = None
model_iv.plda_g = None
model_iv.plda_sigma = None
model_iv.ivectors = None
model_iv.ivectors = training_xv
model_iv.scores = None
model_iv.nb_thread = 1
......@@ -748,7 +747,7 @@ def allies_init_seg(model, model_cfg, show, data_folder, verbose=False):
fh.setLevel(logging.DEBUG)
logger.addHandler(fh)
logger.info(yaml.dump(model_cfg, default_flow_style=False))
#logger.info(yaml.dump(model_cfg, default_flow_style=False))
first_seg_path = f"{model_cfg['tmp_dir']}/seg/first_th{model_cfg['first_seg']['thr_h']}/"
second_seg_path = first_seg_path + f"second_th{model_cfg['within_show']['th_w']}/"
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment