luizapzbn's picture
Upload ./ with huggingface_hub
321c2bb
raw
history blame
2.18 kB
Logging training
Running DummyClassifier()
accuracy: 0.616 average_precision: 0.384 roc_auc: 0.500 recall_macro: 0.500 f1_macro: 0.381
=== new best DummyClassifier() (using recall_macro):
accuracy: 0.616 average_precision: 0.384 roc_auc: 0.500 recall_macro: 0.500 f1_macro: 0.381
Running GaussianNB()
accuracy: 0.425 average_precision: 0.749 roc_auc: 0.825 recall_macro: 0.530 f1_macro: 0.355
=== new best GaussianNB() (using recall_macro):
accuracy: 0.425 average_precision: 0.749 roc_auc: 0.825 recall_macro: 0.530 f1_macro: 0.355
Running MultinomialNB()
accuracy: 0.771 average_precision: 0.800 roc_auc: 0.837 recall_macro: 0.748 f1_macro: 0.752
=== new best MultinomialNB() (using recall_macro):
accuracy: 0.771 average_precision: 0.800 roc_auc: 0.837 recall_macro: 0.748 f1_macro: 0.752
Running DecisionTreeClassifier(class_weight='balanced', max_depth=1)
accuracy: 0.787 average_precision: 0.630 roc_auc: 0.767 recall_macro: 0.767 f1_macro: 0.771
=== new best DecisionTreeClassifier(class_weight='balanced', max_depth=1) (using recall_macro):
accuracy: 0.787 average_precision: 0.630 roc_auc: 0.767 recall_macro: 0.767 f1_macro: 0.771
Running DecisionTreeClassifier(class_weight='balanced', max_depth=5)
accuracy: 0.804 average_precision: 0.801 roc_auc: 0.848 recall_macro: 0.796 f1_macro: 0.794
=== new best DecisionTreeClassifier(class_weight='balanced', max_depth=5) (using recall_macro):
accuracy: 0.804 average_precision: 0.801 roc_auc: 0.848 recall_macro: 0.796 f1_macro: 0.794
Running DecisionTreeClassifier(class_weight='balanced', min_impurity_decrease=0.01)
accuracy: 0.786 average_precision: 0.777 roc_auc: 0.840 recall_macro: 0.773 f1_macro: 0.772
Running LogisticRegression(C=0.1, class_weight='balanced', max_iter=1000)
accuracy: 0.783 average_precision: 0.823 roc_auc: 0.850 recall_macro: 0.779 f1_macro: 0.774
Running LogisticRegression(class_weight='balanced', max_iter=1000)
accuracy: 0.788 average_precision: 0.822 roc_auc: 0.851 recall_macro: 0.784 f1_macro: 0.779
Best model:
DecisionTreeClassifier(class_weight='balanced', max_depth=5)
Best Scores:
accuracy: 0.804 average_precision: 0.801 roc_auc: 0.848 recall_macro: 0.796 f1_macro: 0.794