s470607/ium_470607
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

import tensorflow

Browse Source
This commit is contained in:
jakubknczny 2021-05-15 19:01:47 +02:00
parent 692cfe336b
commit db7a59f689
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
lab5/train/train.py
View File

@ -1,7 +1,7 @@
import numpy as np
import pandas as pd
import sys
import tensorflow as tf
import tensorflow
from tensorflow.keras import layers
X_train = pd.read_csv('train.csv')
@ -17,7 +17,7 @@ Y_test = pd.get_dummies(Y_test)
Y_valid = X_valid.pop('stabf')
Y_valid = pd.get_dummies(Y_valid)
model = tf.keras.Sequential([
model = tensorflow.keras.Sequential([
layers.Input(shape=(12,)),
layers.Dense(32),
layers.Dense(16),
@ -25,12 +25,12 @@ model = tf.keras.Sequential([
])
model.compile(
loss=tf.keras.losses.BinaryCrossentropy(),
optimizer=tf.keras.optimizers.Adam(lr=float(sys.argv[1])),
metrics=[tf.keras.metrics.BinaryAccuracy()])
loss=tensorflow.keras.losses.BinaryCrossentropy(),
optimizer=tensorflow.keras.optimizers.Adam(lr=float(sys.argv[1])),
metrics=[tensorflow.keras.metrics.BinaryAccuracy()])
history = model.fit(tf.convert_to_tensor(X_train, np.float32),
history = model.fit(tensorflow.convert_to_tensor(X_train, np.float32),
Y_train, epochs=2, validation_data=(X_valid, Y_valid))
model.save('grid-stability-dense.h5')