Update 'predict.py'
This commit is contained in:
parent
732babcc90
commit
7968db013b
@ -18,4 +18,6 @@ accu = reloaded.evaluate(x_to_test, y_to_test)
|
||||
pre = reloaded.predict(x_to_test)
|
||||
pre_array = [float(i)/sum(pre) for i in pre]
|
||||
|
||||
pre_array.tofile('prediction.csv', sep=',', format='%s')
|
||||
with open("prediction.txt", "w") as txt_file:
|
||||
for line in pre_array:
|
||||
txt_file.write(f"{line}")
|
Loading…
Reference in New Issue
Block a user