Zaktualizuj 'main.py'
This commit is contained in:
parent
6ecebd9559
commit
200f7b6319
6
main.py
6
main.py
@ -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)
|
Loading…
Reference in New Issue
Block a user