Керас сбрасывает модель или сбрасывает график тензорного потока - PullRequest
0 голосов
/ 01 мая 2018

Моя функция deep_model получает случайные вызовы от другой функции. Каждый раз, когда вызывается функция deep_model, мне нужно сбросить график tf, т.е. я не хочу добавлять слои в существующую модель. Пожалуйста, помогите.

def deep_model(L,N,X_train,y_train,X_test,y_test):
model=models.Sequential()
model.add(layers.Dense(N,activation='relu',input_shape=(4,)))
for i in range(1,L):
    model.add(layers.Dense(N,activation='relu'))
model.add(layers.Dense(1,activation='sigmoid'))
model.compile(optimizer='rmsprop',loss='binary_crossentropy',metrics=['accuracy'])
model.fit(X_train,y_train,epochs=25,batch_size=50)
scores=model.evaluate(X_test, y_test)
return scores[1]

1 Ответ

0 голосов
/ 01 мая 2018

Вы пробовали это?

tf.reset_default_graph()

Вот соответствующая ссылка из документации TF.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...