График работы модели классификатора Keras за эпохи - PullRequest
1 голос
/ 15 апреля 2019

Я использую встроенную оболочку KerasClassifier для выполнения перекрестной проверки kfold в учебной нейронной сети для классификации знаменитого «набора данных Iris».Я хотел бы создать график производительности модели во времени.Я не уверен, как это сделать с помощью оболочки KerasClassifier.

model.history () метод

### Neural Network Time!
import tensorflow as tf 
from tensorflow import keras
from keras.wrappers.scikit_learn import KerasClassifier
from keras.layers import Dense
from keras.models import Sequential
from sklearn.model_selection import KFold

kfold = KFold(n_splits=10, shuffle=True, random_state=seed)

### Build small model
def small_network():
    model = Sequential()
    model.add(Dense(8, activation='relu'))
    model.add(Dense(10, activation='relu'))
    model.add(Dense(3, activation='softmax'))
    # Compile the model
    model.compile(optimizer='adam',
                          loss='categorical_crossentropy',
                          metrics=['accuracy'])
    return model

small_estimator = KerasClassifier(build_fn=small_network, epochs=50, verbose=0)
small_results = cross_val_score(small_estimator, X, y_hot, cv=kfold)
print("Small Network Accuracy: %.2f%% (%.2f%%)" % (small_results.mean()*100, small_results.std()*100))
#GRAPH OF MODEL HISTORY!!!

Я ожидаю, что смогу со временем создать график точности модели с помощьюОбъект KerasClassifier.

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