-
Notifications
You must be signed in to change notification settings - Fork 1.2k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Update the metric module #702
Comments
Or we can convert SINGA Tensor to numpy array and use sklearn.metrics to compute the metrics https://scikit-learn.org/stable/modules/model_evaluation.html. |
When we use this accuracy: so we need a max function to turn the logits into pred label index? |
I suggest to use sklearn's functions to evaluate the metrics. |
sklearn.metrics.accuracy_score API: |
It would be better to implement a metric as a function in the metric.py as metric typically has not states. Therefore, no need to make it as a class. e.g,
Refer to https://keras.io/api/metrics/
The text was updated successfully, but these errors were encountered: