From 7061ff954789c0103a88588a4fb18cb8e289e76c Mon Sep 17 00:00:00 2001 From: Patrice Lopez Date: Sun, 28 Jan 2024 13:03:50 +0100 Subject: [PATCH] debug binary classifier data preparation --- delft/applications/licenseClassifier.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/delft/applications/licenseClassifier.py b/delft/applications/licenseClassifier.py index d45aaaf..e3005d2 100644 --- a/delft/applications/licenseClassifier.py +++ b/delft/applications/licenseClassifier.py @@ -214,7 +214,7 @@ def train_and_eval_binary(embeddings_name, fold_count, architecture="gru", trans y_train_class_rank = np.array(y_train_class_rank) y_test_class_rank = np.array(y_test_class_rank) - list_classes_rank = [list_classes[class_rank], "not_"+list_classes[class_rank]] + list_classes_rank = [list_classes_copyright[class_rank], "not_"+list_classes_copyright[class_rank]] model = Classifier(model_name, architecture=architecture, list_classes=list_classes_rank, max_epoch=max_epoch, fold_number=fold_count, patience=patience, use_roc_auc=True, embeddings_name=embeddings_name, batch_size=batch_size, maxlen=maxlen, early_stop=early_stop, @@ -246,7 +246,7 @@ def train_and_eval_binary(embeddings_name, fold_count, architecture="gru", trans y_train_class_rank = np.array(y_train_class_rank) y_test_class_rank = np.array(y_test_class_rank) - list_classes_rank = [list_classes[class_rank], "not_"+list_classes[class_rank]] + list_classes_rank = [list_classes_licenses[class_rank], "not_"+list_classes_licenses[class_rank]] model = Classifier(model_name, architecture=architecture, list_classes=list_classes_rank, max_epoch=max_epoch, fold_number=fold_count, patience=patience, use_roc_auc=True, embeddings_name=embeddings_name, batch_size=batch_size, maxlen=maxlen, early_stop=early_stop,