Model fix
This commit is contained in:
parent
90ea83a9f0
commit
f1f3d2647a
@ -26,7 +26,7 @@ def get_x_y(data):
|
|||||||
|
|
||||||
|
|
||||||
def predict():
|
def predict():
|
||||||
model = load_model('baltimore_model3')
|
model = load_model('baltimore_model.h5')
|
||||||
|
|
||||||
train = pd.read_csv('baltimore_train.csv')
|
train = pd.read_csv('baltimore_train.csv')
|
||||||
baltimore_data_test = pd.read_csv('baltimore_test.csv')
|
baltimore_data_test = pd.read_csv('baltimore_test.csv')
|
||||||
|
@ -48,7 +48,7 @@ def train_model():
|
|||||||
validation_split=0.2)
|
validation_split=0.2)
|
||||||
hist = pd.DataFrame(history.history)
|
hist = pd.DataFrame(history.history)
|
||||||
hist['epoch'] = history.epoch
|
hist['epoch'] = history.epoch
|
||||||
model.save('baltimore_model3')
|
model.save('baltimore_model.h5')
|
||||||
|
|
||||||
|
|
||||||
train_model()
|
train_model()
|
||||||
|
Loading…
Reference in New Issue
Block a user