keras added

2021-05-07 19:11:51 +02:00 · 2021-05-07 19:11:51 +02:00 · 71fba52c32
commit 71fba52c32
parent 367ec7d511
2 changed files with 2 additions and 0 deletions
--- a/avocado-training.py
+++ b/avocado-training.py
@ -5,6 +5,7 @@ from sklearn import preprocessing
 from sklearn.linear_model import LinearRegression
 from sklearn.metrics import mean_squared_error
 import tensorflow as tf
+from tensorflow import keras
 from tensorflow.keras.layers import Input, Dense, Activation,Dropout
 from tensorflow.keras.models import Model
 from tensorflow.keras.callbacks import EarlyStopping
--- a/requirements.txt
+++ b/requirements.txt
@ -3,5 +3,6 @@ pandas==1.2.4
 numpy==1.19.2
 sklearn
 tensorflow==2.4.1
+keras
 jinja2==2.11.3
 matplotlib