Keras-模型干扰

时间:2018-11-03 01:54:16

标签: python keras deep-learning

我正在Django中进行开发,并且在两个不同的视图中有两个模型。一种模型与另一种模型无关,但是又被称为另一种。

模型1:

    classificador = Sequential()
    classificador.add(Dense(units=60, activation="relu", kernel_initializer="random_uniform", input_shape=(38,)))
    classificador.add(Dense(units=60, activation="relu", kernel_initializer="random_uniform"))
    classificador.add(Dense(units=1, activation="sigmoid"))
    classificador.compile(optimizer="adam", loss="binary_crossentropy", metrics=["binary_accuracy"])
    classificador.fit(X_Treino, Y_Treino, batch_size=10, epochs=100)

模型2

    classificador_outlier = Sequential()
    classificador_outlier.add(Dense(units=45, activation="relu", kernel_initializer="random_uniform", input_shape=(6,)))
    classificador_outlier.add(Dense(units=45, activation="relu", kernel_initializer="random_uniform"))
    classificador_outlier.add(Dense(units=1, activation="sigmoid"))
    classificador_outlier.compile(optimizer="adam", loss="binary_crossentropy", metrics=["binary_accuracy"])
    classificador_outlier.fit(X_Treino_Outlier, Y_Treino_Outlier, batch_size=10, epochs=100)

我认为一个模型正在干扰另一个模型。如果我只运行一个模型(任何人),它将很好地工作。两者同时运行时,我收到以下消息,运行Model 1(最后一个运行):

  

与矩阵大小不兼容:In [0]:[10,38],In [1]:[6,45]

观察到Model 1的输入具有(10,38)形状,并且消息错误是与Model 2相关的参考值。

那会是什么?

1 个答案:

答案 0 :(得分:0)

清除会话后工作正常,如here所示

from keras import backend as K
K.clear_session()