Update 'ium_zadanie6_training.py'
This commit is contained in:
parent
516e90125b
commit
2e39677b03
@ -1,7 +1,8 @@
|
||||
import sys
|
||||
from tensorflow.keras import layers
|
||||
from tensorflow.keras import layers, Sequential
|
||||
# from keras.layers import Flatten,Dense,Dropout, GlobalAveragePooling2D
|
||||
from tensorflow.keras.optimizers import Adam
|
||||
from tensorflow import convert_to_tensor
|
||||
import numpy as np
|
||||
import pandas as pd
|
||||
from sklearn.metrics import mean_squared_error
|
||||
@ -14,7 +15,7 @@ x_train.pop('Unnamed: 0')
|
||||
|
||||
learning_rate = 0.001 #sys.argv[1]
|
||||
|
||||
model = tensorflow.keras.Sequential()
|
||||
model = Sequential()
|
||||
model.add(layers.Input(shape=(22,)))
|
||||
model.add(layers.Dense(64))
|
||||
model.add(layers.Dense(64))
|
||||
@ -24,7 +25,7 @@ model.add(layers.Dense(1))
|
||||
model.compile(loss='mean_absolute_error', optimizer=Adam(learning_rate))
|
||||
|
||||
model.fit(
|
||||
x = tensorflow.convert_to_tensor(x_train, np.float32),
|
||||
x = convert_to_tensor(x_train, np.float32),
|
||||
y = y_train,
|
||||
verbose=0, epochs=99)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user