|
1 | | -import json |
2 | | -from delft.utilities.Utilities import split_data_and_labels, t_or_f |
3 | | -from delft.textClassification.reader import load_dataseer_corpus_csv |
4 | | -from delft.textClassification.reader import vectorize as vectorizer |
5 | | -from delft.textClassification import Classifier |
6 | 1 | import argparse |
| 2 | +import json |
7 | 3 | import time |
8 | | -from delft.textClassification.models import architectures |
| 4 | + |
9 | 5 | import numpy as np |
10 | 6 |
|
| 7 | +from delft.textClassification import Classifier |
| 8 | +from delft.textClassification.models import architectures |
| 9 | +from delft.textClassification.reader import load_dataseer_corpus_csv |
| 10 | +from delft.textClassification.reader import vectorize as vectorizer |
| 11 | +from delft.utilities.Utilities import split_data_and_labels, t_or_f |
| 12 | + |
11 | 13 | """ |
12 | | - Classifier for deciding if a sentence introduce a dataset or not, and prediction of the |
13 | | - dataset type. |
| 14 | + Classifier for deciding if a sentence introduce a dataset or not, and prediction of the |
| 15 | + dataset type. |
14 | 16 | """ |
15 | 17 |
|
16 | 18 |
|
@@ -198,8 +200,8 @@ def train( |
198 | 200 |
|
199 | 201 | model_name = 'dataseer-' + the_class + "_" + architecture |
200 | 202 |
|
201 | | - model = Classifier(model_name, architecture=architecture, list_classes=datatypes_list_subclasses[the_class], max_epoch=max_epoch, |
202 | | - fold_number=fold_count, patience=patience, use_roc_auc=True, embeddings_name=embeddings_name, |
| 203 | + model = Classifier(model_name, architecture=architecture, list_classes=datatypes_list_subclasses[the_class], max_epoch=max_epoch, |
| 204 | + fold_number=fold_count, patience=patience, use_roc_auc=True, embeddings_name=embeddings_name, |
203 | 205 | batch_size=batch_size, class_weights=class_weights, early_stop=early_stop, transformer_name=transformer) |
204 | 206 |
|
205 | 207 | if fold_count == 1: |
@@ -560,7 +562,6 @@ def train_eval_cascaded(embeddings_name, fold_count, architecture="gru", transfo |
560 | 562 | model_binary.eval(x_test, y_test) |
561 | 563 |
|
562 | 564 | x_test_binary = x_test |
563 | | - y_test_binary = y_test |
564 | 565 |
|
565 | 566 | # second, the first level datatype taxonomy for sentences classified as dataset |
566 | 567 | ( |
@@ -610,7 +611,6 @@ def train_eval_cascaded(embeddings_name, fold_count, architecture="gru", transfo |
610 | 611 |
|
611 | 612 | # eval by cascading |
612 | 613 | result_binary = model_binary.predict(x_test_binary, output_format="default") |
613 | | - result_first = model_first.predict(x_test, output_format="default") |
614 | 614 |
|
615 | 615 | # select sequences classified as dataset |
616 | 616 | result_intermediate = np.asarray([np.argmax(line) for line in result_binary]) |
@@ -810,7 +810,7 @@ def build_prior_class_distribution(): |
810 | 810 | if architecture not in architectures: |
811 | 811 | print("unknown model architecture, must be one of " + str(architectures)) |
812 | 812 |
|
813 | | - if transformer == None and embeddings_name == None: |
| 813 | + if transformer is None and embeddings_name is None: |
814 | 814 | # default word embeddings |
815 | 815 | embeddings_name = "glove-840B" |
816 | 816 |
|
|
0 commit comments