Skip to content

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

add ModelCheckpoint

parent 80a61c62
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
backend/IT18218640/train.py
View file @ f1ca7dff
...@@ -3,6 +3,7 @@ import numpy as np ...@@ -3,6 +3,7 @@ import numpy as np
import tensorflow.keras as keras import tensorflow.keras as keras
from sklearn.model_selection import train_test_split from sklearn.model_selection import train_test_split
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from keras.callbacks import ModelCheckpoint
DATA_PATH = "data.json" DATA_PATH = "data.json"
SAVE_MODEL_PATH = "model.h5" 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