This commit is contained in:
Zofia Galla 2021-05-16 22:54:11 +02:00
parent 197efd7bcb
commit feddcf2b36

View File

@ -33,7 +33,6 @@ def prepare_train_model(learning_rate, _run):
y_train = x_train.pop('rottentomatoes_audience_score') y_train = x_train.pop('rottentomatoes_audience_score')
x_train.pop('Unnamed: 0') x_train.pop('Unnamed: 0')
learning_rate = float(sys.argv[1])
model = Sequential() model = Sequential()
model.add(layers.Input(shape=(22,))) model.add(layers.Input(shape=(22,)))
model.add(layers.Dense(64)) model.add(layers.Dense(64))