Commit 8728290d authored by Noric Couderc's avatar Noric Couderc

Fixed saving the names of features

The feature names were not actually saved originally.
parent 104e7937
......@@ -358,17 +358,19 @@ if __name__ == "__main__":
sw_hw_cleaned["Sample weight"],
# Fitting without the sample weights gives better
# Accuracy than with it, not sure why.
classifier = RandomForestClassifier(), y_train, w_train)
# We just add it to the class
classifier.feature_names = poly_transformer.get_feature_names()
classifier.feature_names = poly_transformer \
print("Accuracy: {0}".format(classifier.score(X_test, y_test, w_test)))
print("Classifier trained")
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