I’m using Flux to implement a handwritten digit classifier based on the MNIST dataset (ex https://github.com/FluxML/model-zoo/blob/master/vision/mnist/mlp.jl ). Is there any way to compute the confusion matrix in order to evaluate the performance of the model (need precision, recall and F1 score). T…

See Performance Measures · MLJ (alan-turing-institute.github.io) julia> using CategoricalArrays, MLJ julia> yhat = rand(1:10, 100)|>CategoricalArray; julia> y = rand(1:10, 100)|>CategoricalArray; julia> ConfusionMatrix()(yhat, y) ┌ Warning: The classes are un-ordered, │ using order: [1, 2, 3, 4,…

Flux.jl confusion matrix

General Usage

zgornel February 4, 2020, 10:05am 10

I guess so as it is part of JuliaStats yet it receives very little attention. I would not really rely on it.

Topic		Replies	Views
Function `confusion_matrix ` in MLJ always fails General Usage mlj	4	670	May 23, 2022
MLJ confusion_matrix() - MethodError Machine Learning question , package	5	1338	September 18, 2020
Flux, categorical arrays, roc curves, confusion matrices Machine Learning flux	14	1108	December 12, 2022
Article that explains using Flux on MNIST from the model-zoo example Machine Learning	4	3274	October 27, 2018
Introductory ML model in Flux General Usage	2	385	May 10, 2021

Flux.jl confusion matrix

Related topics