Skip to content
Snippets Groups Projects
Commit 5b6e11c6 authored by ym1r22's avatar ym1r22
Browse files

Update RandomForest2.py

parent 844f143c
No related branches found
No related tags found
No related merge requests found
...@@ -52,7 +52,7 @@ print(y_pred) ...@@ -52,7 +52,7 @@ print(y_pred)
# Save the test predictions to a CSV file # Save the test predictions to a CSV file
test_data["PredictedLabel"] = y_pred test_data["PredictedLabel"] = y_pred
test_data.to_csv("TestPrediction2.csv", index=False) test_data.to_csv("TestingResultsMulti.csv", index=False)
# Calculate the confusion matrix # Calculate the confusion matrix
confusion_mtx = confusion_matrix(y_test, y_test_pred) confusion_mtx = confusion_matrix(y_test, y_test_pred)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment