s434742/ium_434742
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

keras added
All checks were successful
s434742-training/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
patrycjalazna 2021-05-07 19:11:51 +02:00
parent 367ec7d511
commit 71fba52c32
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
avocado-training.py
View File

@ -5,6 +5,7 @@ from sklearn import preprocessing
from sklearn.linear_model import LinearRegression from sklearn.linear_model import LinearRegression
from sklearn.metrics import mean_squared_error from sklearn.metrics import mean_squared_error
import tensorflow as tf import tensorflow as tf
from tensorflow import keras
from tensorflow.keras.layers import Input, Dense, Activation,Dropout from tensorflow.keras.layers import Input, Dense, Activation,Dropout
from tensorflow.keras.models import Model from tensorflow.keras.models import Model
from tensorflow.keras.callbacks import EarlyStopping from tensorflow.keras.callbacks import EarlyStopping

1
requirements.txt
View File

@ -3,5 +3,6 @@ pandas==1.2.4
numpy==1.19.2 numpy==1.19.2
sklearn sklearn
tensorflow==2.4.1 tensorflow==2.4.1
keras
jinja2==2.11.3 jinja2==2.11.3
matplotlib matplotlib