Skip to content

21_22-J 38
21_22-J 38
Commit 203f9019 authored by W.D.R.P. Sandeepa's avatar W.D.R.P. Sandeepa
Browse files
Options

Merge branch 'it18218640' into 'master' 

add ModelCheckpoint

See merge request !96
parents 80a61c62 f1ca7dff
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
backend/IT18218640/train.py
View file @ 203f9019
......@@ -3,6 +3,7 @@ import numpy as np
import tensorflow.keras as keras
from sklearn.model_selection import train_test_split
import matplotlib.pyplot as plt
from keras.callbacks import ModelCheckpoint
DATA_PATH = "data.json"
SAVE_MODEL_PATH = "model.h5"
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment