파이썬 Keras에서 정밀도 및 재현율을 계산하는 방법

Keras 2.02 (Tensorflow 백엔드 포함)로 다중 클래스 분류기를 만들고 있으며 Keras에서 정밀도와 재현율을 계산하는 방법을 모릅니다. 제발 도와주세요.

 

해결 방법

 

import keras
import keras_metrics

model = models.Sequential()
model.add(keras.layers.Dense(1, activation="sigmoid", input_dim=2))
model.add(keras.layers.Dense(1, activation="softmax"))

model.compile(optimizer="sgd",
              loss="binary_crossentropy",
              metrics=[keras_metrics.precision(), keras_metrics.recall()])

업데이트 : Keras 버전 2.3.0 부터 정밀도, 재현율 등과 같은 측정 항목이 라이브러리 배포 패키지에 제공됩니다.

사용법은 다음과 같습니다.

model.compile(optimizer="sgd",
              loss="binary_crossentropy",
              metrics=[keras.metrics.Precision(), keras.metrics.Recall()])

 

참조 페이지 https://stackoverflow.com/questions/43076609