Chcę obliczyć precyzję, pamięć i wynik F1 dla mojego binarnego modelu KerasClassifier, ale nie znajduję żadnego rozwiązania.
Oto mój rzeczywisty kod:
# Split dataset in train and test data
X_train, X_test, Y_train, Y_test = train_test_split(normalized_X, Y, test_size=0.3, random_state=seed)
# Build the model
model = Sequential()
model.add(Dense(23, input_dim=45, kernel_initializer='normal', activation='relu'))
model.add(Dense(1, kernel_initializer='normal', activation='sigmoid'))
# Compile model
model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
tensorboard = TensorBoard(log_dir="logs/{}".format(time.time()))
time_callback = TimeHistory()
# Fit the model
history = model.fit(X_train, Y_train, validation_split=0.3, epochs=200, batch_size=5, verbose=1, callbacks=[tensorboard, time_callback])
A potem przewiduję nowe dane testowe i otrzymuję macierz zamieszania w następujący sposób:
y_pred = model.predict(X_test)
y_pred =(y_pred>0.5)
list(y_pred)
cm = confusion_matrix(Y_test, y_pred)
print(cm)
Ale czy jest jakieś rozwiązanie, aby uzyskać wynik dokładności, wynik F1, precyzję i wycofanie? (Jeśli nie jest to skomplikowane, również ocena krzyżowa, ale nie jest to konieczne dla tej odpowiedzi)
Dziękuję za wszelką pomoc!
Możesz użyć raportu klasyfikacji scikit-learn . Aby przekonwertować etykiety na format numeryczny lub binarny, spójrz na koder etykiet scikit-learn .
co daje (dane wyjściowe skopiowane z przykładu scikit-learn):
źródło
Możesz także spróbować, jak wspomniano poniżej.
źródło
Spróbuj tego: https://scikit-learn.org/stable/modules/generated/sklearn.metrics.precision_recall_fscore_support.html z Y_test, y_pred jako parametrami.
źródło