Zaktualizuj 'main.py'

This commit is contained in:
Jerzy Kwiatkowski 2023-06-23 09:12:07 +02:00
parent 6ecebd9559
commit 200f7b6319

View File

@ -25,9 +25,5 @@ classifier = RandomForestClassifier()
classifier.fit(X_train, y_transformed)
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.to_csv('prediction', sep=' ', index=False)
data_test.to_csv('prediction.dat', sep=' ', index=False)