Юнит тест с керасом - PullRequest
Новая
       12

Юнит тест с керасом

1 голос
/ 22 июня 2019

У меня есть приложение для машинного обучения, построенное поверх Keras. Я написал такие методы, как normalize_dataset, get_model, train_model, predict_class и так далее. Теперь я хочу написать модульный тест для этого метода, чтобы убедиться, что они работают правильно, но я не знаю, как это сделать.

Пример: 

def get_model():
    model = Sequential()
    model.add(Dense(32, activation='relu', input_dim=100))
    model.add(Dense(1, activation='sigmoid'))
    model.compile(optimizer='rmsprop',
                  loss='binary_crossentropy',
                  metrics=['accuracy'])
    return model

здесь я хочу написать тест для этого метода, чтобы быть уверенным, что когда я добавлю еще слой в свою модель, он будет работать так же хорошо, как и сейчас.

Обновление: очень похоже на первый пример этой темы , но через Keras.

...