Update 'predict.py'
This commit is contained in:
parent
433b6b6c0b
commit
bc26ba7528
@ -16,8 +16,5 @@ 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]
|
|
||||||
|
|
||||||
with open("prediction.txt", "w") as txt_file:
|
pre.tofile('prediction.csv', sep=',', format='%s')
|
||||||
for line in pre_array:
|
|
||||||
txt_file.write(f"{line}")
|
|
Loading…
Reference in New Issue
Block a user