Code

Add precision classification metric#2293

Code

Merged

laggui merged 31 commits intotracel-ai:maintracel-ai/burn:mainfrom tsanona:add-to-metricstsanona/burn:add-to-metricsCopy head branch name to clipboardNov 20, 2024

+694-1

Commits

Commits on Sep 21, 2024

• Implement confusion matrix and precision, first draft

  

  Tiago Sanona

  committed
  c73438f

  Copy full SHA for c73438f

• Implement confusion matrix

  

  Tiago Sanona

  committed
  63f4a1d

  Copy full SHA for 63f4a1d

• format :D

  

  Tiago Sanona

  committed
  b9d71b6

  Copy full SHA for b9d71b6

• add agg type to cm, reformat debug representation add testing.

  Show description for eac29aa
  

  Tiago Sanona

  committed
  eac29aa

  Copy full SHA for eac29aa

• formating and tiny refactor

  

  Tiago Sanona

  committed
  59db68b

  Copy full SHA for 59db68b

• add ClassificationMetric trait, rename variables and types, move test module to lib.rs make precision a classification metric.

  

  Tiago Sanona

  committed
  4261bd8

  Copy full SHA for 4261bd8

• change unwrap to expect

  

  Tiago Sanona

  committed
  5431a2f

  Copy full SHA for 5431a2f

• update book

  

  Tiago Sanona

  committed
  fd2e585

  Copy full SHA for fd2e585

Commits on Sep 22, 2024

• remove unused code

  

  Tiago Sanona

  committed
  56965e8

  Copy full SHA for 56965e8

• changes to make reusing code easier

  

  Tiago Sanona

  committed
  419438a

  Copy full SHA for 419438a

• format :D

  

  Tiago Sanona

  committed
  dfac847

  Copy full SHA for dfac847

Commits on Sep 24, 2024

• change to static data tests

  

  Tiago Sanona

  committed
  ea4b29c

  Copy full SHA for ea4b29c

Commits on Oct 14, 2024

• remove classification metric trait, add auxiliary code for classification input, clarify descriptions, remove dead code, rename some objects

  

  Tiago Sanona

  committed
  e23aa7b

  Copy full SHA for e23aa7b

Commits on Oct 21, 2024

• move classification objects to classification.rs, use rstest, remove approx lib and use tensordata asserts, move aggregate and average functions to ConfusionStats implementation

  

  Tiago Sanona

  committed
  60a246b

  Copy full SHA for 60a246b

Commits on Oct 23, 2024

• review docstring, add top_k for multiclass tasks.

  

  Tiago Sanona

  committed
  c145531

  Copy full SHA for c145531

Commits on Oct 25, 2024

• move class averaging and metric computation to metric implementation, make dummy data more predictable and add tests for top_k > 1

  

  Tiago Sanona

  committed
  0c984c4

  Copy full SHA for 0c984c4

• change struct and var names

  

  Tiago Sanona

  committed
  b0a2939

  Copy full SHA for b0a2939

Commits on Oct 26, 2024

• Merge branch 'main' into add-to-metrics

  

  Tiago Sanona

  committed
  f18e321

  Copy full SHA for f18e321

Commits on Oct 30, 2024

• rename params, enforce nonzero for top_k param, optimize one_hot for case num_class = 1, reformat dummy data, make use of derive(new) for metric init.

  

  Tiago Sanona

  committed
  386802c

  Copy full SHA for 386802c

Commits on Nov 1, 2024

• add adaptor por classification input, correct one hot function

  

  Tiago Sanona

  committed
  b525527

  Copy full SHA for b525527

Commits on Nov 8, 2024

• define default for ClassReduction, derive new for Precision metric with class_reduction as default and new setter implementation, move NonZerousize boundary to confusion_stats

  

  Tiago Sanona

  committed
  ff7611a

  Copy full SHA for ff7611a

• Merge branch 'main' into add-to-metrics

  

  Tiago Sanona

  committed
  4cbcff2

  Copy full SHA for 4cbcff2

• expose PrecisionMetric, change metric initialization

  

  Tiago Sanona

  committed
  eeab0d3

  Copy full SHA for eeab0d3

Commits on Nov 16, 2024

• check one_hot input tensor has more than 1 classes and correct it's implementation, deal with classification output with 1 class, make macro average default, expose ClassReduction type and split pr…

  Show description for aea207f
  

  Tiago Sanona

  committed
  aea207f

  Copy full SHA for aea207f

• Merge branch 'main' into add-to-metrics

  Show description for 410f273
  

  Tiago Sanona

  committed
  410f273

  Copy full SHA for 410f273

• implement adaptor for MultilabelClassificationOutput and ClassificationInput

  

  Tiago Sanona

  committed
  746fa9d

  Copy full SHA for 746fa9d

Commits on Nov 18, 2024

• change with_top_k to take usize

  

  Tiago Sanona

  committed
  7428b86

  Copy full SHA for 7428b86

• Merge branch 'main' into add-to-metrics

  

  Tiago Sanona

  committed
  58e1902

  Copy full SHA for 58e1902

• Add precision config for binary, multiclass and multilabel

  laggui

  committed
  d598f00

  Copy full SHA for d598f00

• Fix dummy_classification_input

  laggui

  committed
  1542ee9

  Copy full SHA for 1542ee9

Commits on Nov 19, 2024

• make PrecisionMetric public

  

  Tiago Sanona

  committed
  03ebe1d

  Copy full SHA for 03ebe1d