s464967/Inzynierka
2
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Zaktualizuj 'main.py'

Browse Source
This commit is contained in:
Jerzy Kwiatkowski 2023-06-23 09:12:07 +02:00
parent 6ecebd9559
commit 200f7b6319
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
main.py
View File

@ -25,9 +25,5 @@ classifier = RandomForestClassifier()
classifier.fit(X_train, y_transformed) classifier.fit(X_train, y_transformed)
y_pred = classifier.predict(X_test) y_pred = classifier.predict(X_test)
# for i, prediction in enumerate(y_pred):
# print(data_test.iloc[i, 0] + ": ")
# print(prediction)
data_test['8'] = list(y_pred) data_test['8'] = list(y_pred)
data_test.to_csv('prediction', sep=' ', index=False) data_test.to_csv('prediction.dat', sep=' ', index=False)