keras重置模型或重置张量流图

Question

我的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]

Answer 1

你试过这个吗？

tf.reset_default_graph()

以下是各自的link from the TF Documentation.