Commit dccab9e4 authored by ALVES Guilherme's avatar ALVES Guilherme
Browse files

Printing accuracy

parent fcc583c4
......@@ -30,7 +30,7 @@ def main(source_name, train_size, to_drop, all_categorical_features, max_feature
threshold_1 = find_threshold(original_model.prob(train), labels_train)
accuracy = evaluation(original_model.prob(test), labels_test, threshold_1)
print(accuracy)
print("Original model accuracy:",accuracy)
is_fair ,_ = exp(original_model, train, max_features, to_drop, feature_names, class_names, all_categorical_features, categorical_names)
......@@ -44,7 +44,7 @@ def main(source_name, train_size, to_drop, all_categorical_features, max_feature
threshold_2 = find_threshold(ensemble.prob(train), labels_train)
accuracy = evaluation(ensemble.prob(test), labels_test, threshold_2)
print(accuracy)
print("ExpOut ensemble's model accuracy:", accuracy)
is_fair ,_ = exp(ensemble, train, max_features, to_drop, feature_names, class_names, all_categorical_features, categorical_names)
return ensemble
......
Markdown is supported
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