Update 'predict.py'
This commit is contained in:
parent
312735d7ba
commit
0445bc9d82
@ -7,7 +7,7 @@ from sklearn.model_selection import train_test_split
|
||||
import tensorflow
|
||||
|
||||
|
||||
reloaded = tensorflow.keras.models.load_model('.')
|
||||
reloaded = tensorflow.keras.models.load_model('saved_model.pb')
|
||||
|
||||
x_to_test = pandas.read_csv('./X_test.csv')
|
||||
y_to_test = pandas.read_csv('./Y_test.csv')
|
||||
|
Loading…
Reference in New Issue
Block a user