add conf and training results

_best/dev.yaml

@@ -0,0 +1,27 @@
+ml-accuracy:
+  Anth: 0.9706353443673251
+  Bio: 0.9508809396689802
+  Geo: 0.8510411105178858
+  Sil: 0.9583555792845702
+  all: 0.9327282434596903
+ml-f1-micro:
+  Anth: 0.9706353443673251
+  Bio: 0.9508809396689802
+  Geo: 0.8510411105178857
+  Sil: 0.9583555792845702
+  all: 0.9327282434596903
+ml-f1-macro:
+  Anth: 0.9701027011323622
+  Bio: 0.9508617641355553
+  Geo: 0.8394133823871512
+  Sil: 0.8236346516007533
+  all: 0.8960031248139555
+ml-f1-weighted:
+  Anth: 0.9706119080649868
+  Bio: 0.9508871587609018
+  Geo: 0.8468652795820465
+  Sil: 0.9547344914299045
+  all: 0.93077470945946
+dev_loss:
+  all: 0.18468653024007967
+iteration: 10

_best/model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:506016b0dee0ba973eb54f7797f6b28497ed65d02747a9d0aa522042fb21fc7b
+size 19835377

_best/optimizer.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0353f8f3cd590c22556b873e221711bf165057e704cb221eb3c8d52848e8802e
+size 39633220

_test/test_holistic.yaml

@@ -0,0 +1,26 @@
+ml-accuracy:
+  Anth: 0.9853528628495339
+  Bio: 0.9427430093209055
+  Geo: 0.7696404793608522
+  Sil: 0.9116733244562805
+  all: 0.902352418996893
+ml-f1-micro:
+  Anth: 0.9853528628495338
+  Bio: 0.9427430093209055
+  Geo: 0.7696404793608522
+  Sil: 0.9116733244562805
+  all: 0.902352418996893
+ml-f1-macro:
+  Anth: 0.9769561143673973
+  Bio: 0.937078388485209
+  Geo: 0.7695603976552177
+  Sil: 0.803393227139678
+  all: 0.8717470319118755
+ml-f1-weighted:
+  Anth: 0.9854084295674305
+  Bio: 0.9426173150715926
+  Geo: 0.769005545837607
+  Sil: 0.9028010916019895
+  all: 0.8999580955196549
+loss:
+  all: 0.29107842086420826

conf/config.yaml

@@ -0,0 +1,17 @@
+defaults:
+  - _autrainer_
+  - _self_
+
+results_dir: results
+experiment_id: default
+iterations: 30
+
+hydra:
+  sweeper:
+    params:
+      +seed: 1
+      +batch_size: 16
+      +learning_rate: 0.001
+      dataset: EDANSA2019-32k
+      model: Cnn10-32k-T
+      optimizer: Adam

file_handler.yaml

@@ -0,0 +1 @@
+$autrainer.datasets.utils.file_handlers.NumpyFileHandler==0.2.0: {}

inference_transform.yaml

@@ -0,0 +1,2 @@
+$autrainer.transforms.smart_compose.SmartCompose==0.2.0:
+  transforms: []

model.yaml

@@ -0,0 +1,7 @@
+$autrainer.models.cnn_10.Cnn10==0.2.0:
+  output_dim: 4
+  sigmoid_output: false
+  sigmoid_predictions: false
+  segmentwise: false
+  in_channels: 1
+  transfer: https://zenodo.org/records/3987831/files/Cnn10_mAP%3D0.380.pth

target_transform.yaml

@@ -0,0 +1,7 @@
+$autrainer.datasets.utils.target_transforms.MultiLabelEncoder==0.2.0:
+  threshold: 0.5
+  labels:
+  - Anth
+  - Bio
+  - Geo
+  - Sil