From 71fba52c327bee3761d6075ba574088c0ab4de92 Mon Sep 17 00:00:00 2001
From: patrycjalazna <patrycja.lazna@gmail.com>
Date: Fri, 7 May 2021 19:11:51 +0200
Subject: [PATCH] keras added

---
 avocado-training.py | 1 +
 requirements.txt    | 1 +
 2 files changed, 2 insertions(+)

diff --git a/avocado-training.py b/avocado-training.py
index 2211db7..1f71295 100644
--- 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
diff --git a/requirements.txt b/requirements.txt
index 454e683..f9d5201 100644
--- 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
\ No newline at end of file