Gitlab service upgraded. Please report any problems to

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