Fix lr
This commit is contained in:
parent
197efd7bcb
commit
feddcf2b36
@ -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))
|
||||||
|
Loading…
Reference in New Issue
Block a user