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 = reloaded.predict(x_to_test)
|
||||||
pre_array = [float(i)/sum(pre) for i in pre]
|
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