Mercurial > repos > goeckslab > image_learner
annotate constants.py @ 3:2c3a3dfaf1a9 draft default tip
planemo upload for repository https://github.com/goeckslab/gleam.git commit 5ab02b6688c9af85525dd225134db8f2bbed76ed
author | goeckslab |
---|---|
date | Fri, 04 Jul 2025 03:44:56 +0000 |
parents | 186424a7eca7 |
children |
rev | line source |
---|---|
2
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
1 from typing import Any, Dict |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
2 |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
3 # --- Constants --- |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
4 SPLIT_COLUMN_NAME = "split" |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
5 LABEL_COLUMN_NAME = "label" |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
6 IMAGE_PATH_COLUMN_NAME = "image_path" |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
7 DEFAULT_SPLIT_PROBABILITIES = [0.7, 0.1, 0.2] |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
8 TEMP_CSV_FILENAME = "processed_data_for_ludwig.csv" |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
9 TEMP_CONFIG_FILENAME = "ludwig_config.yaml" |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
10 TEMP_DIR_PREFIX = "ludwig_api_work_" |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
11 MODEL_ENCODER_TEMPLATES: Dict[str, Any] = { |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
12 "stacked_cnn": "stacked_cnn", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
13 "resnet18": {"type": "resnet", "model_variant": 18}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
14 "resnet34": {"type": "resnet", "model_variant": 34}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
15 "resnet50": {"type": "resnet", "model_variant": 50}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
16 "resnet101": {"type": "resnet", "model_variant": 101}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
17 "resnet152": {"type": "resnet", "model_variant": 152}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
18 "resnext50_32x4d": {"type": "resnext", "model_variant": "50_32x4d"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
19 "resnext101_32x8d": {"type": "resnext", "model_variant": "101_32x8d"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
20 "resnext101_64x4d": {"type": "resnext", "model_variant": "101_64x4d"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
21 "resnext152_32x8d": {"type": "resnext", "model_variant": "152_32x8d"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
22 "wide_resnet50_2": {"type": "wide_resnet", "model_variant": "50_2"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
23 "wide_resnet101_2": {"type": "wide_resnet", "model_variant": "101_2"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
24 "wide_resnet103_2": {"type": "wide_resnet", "model_variant": "103_2"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
25 "efficientnet_b0": {"type": "efficientnet", "model_variant": "b0"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
26 "efficientnet_b1": {"type": "efficientnet", "model_variant": "b1"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
27 "efficientnet_b2": {"type": "efficientnet", "model_variant": "b2"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
28 "efficientnet_b3": {"type": "efficientnet", "model_variant": "b3"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
29 "efficientnet_b4": {"type": "efficientnet", "model_variant": "b4"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
30 "efficientnet_b5": {"type": "efficientnet", "model_variant": "b5"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
31 "efficientnet_b6": {"type": "efficientnet", "model_variant": "b6"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
32 "efficientnet_b7": {"type": "efficientnet", "model_variant": "b7"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
33 "efficientnet_v2_s": {"type": "efficientnet", "model_variant": "v2_s"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
34 "efficientnet_v2_m": {"type": "efficientnet", "model_variant": "v2_m"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
35 "efficientnet_v2_l": {"type": "efficientnet", "model_variant": "v2_l"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
36 "regnet_y_400mf": {"type": "regnet", "model_variant": "y_400mf"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
37 "regnet_y_800mf": {"type": "regnet", "model_variant": "y_800mf"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
38 "regnet_y_1_6gf": {"type": "regnet", "model_variant": "y_1_6gf"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
39 "regnet_y_3_2gf": {"type": "regnet", "model_variant": "y_3_2gf"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
40 "regnet_y_8gf": {"type": "regnet", "model_variant": "y_8gf"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
41 "regnet_y_16gf": {"type": "regnet", "model_variant": "y_16gf"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
42 "regnet_y_32gf": {"type": "regnet", "model_variant": "y_32gf"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
43 "regnet_y_128gf": {"type": "regnet", "model_variant": "y_128gf"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
44 "regnet_x_400mf": {"type": "regnet", "model_variant": "x_400mf"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
45 "regnet_x_800mf": {"type": "regnet", "model_variant": "x_800mf"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
46 "regnet_x_1_6gf": {"type": "regnet", "model_variant": "x_1_6gf"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
47 "regnet_x_3_2gf": {"type": "regnet", "model_variant": "x_3_2gf"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
48 "regnet_x_8gf": {"type": "regnet", "model_variant": "x_8gf"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
49 "regnet_x_16gf": {"type": "regnet", "model_variant": "x_16gf"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
50 "regnet_x_32gf": {"type": "regnet", "model_variant": "x_32gf"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
51 "vgg11": {"type": "vgg", "model_variant": 11}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
52 "vgg11_bn": {"type": "vgg", "model_variant": "11_bn"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
53 "vgg13": {"type": "vgg", "model_variant": 13}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
54 "vgg13_bn": {"type": "vgg", "model_variant": "13_bn"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
55 "vgg16": {"type": "vgg", "model_variant": 16}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
56 "vgg16_bn": {"type": "vgg", "model_variant": "16_bn"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
57 "vgg19": {"type": "vgg", "model_variant": 19}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
58 "vgg19_bn": {"type": "vgg", "model_variant": "19_bn"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
59 "shufflenet_v2_x0_5": {"type": "shufflenet_v2", "model_variant": "x0_5"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
60 "shufflenet_v2_x1_0": {"type": "shufflenet_v2", "model_variant": "x1_0"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
61 "shufflenet_v2_x1_5": {"type": "shufflenet_v2", "model_variant": "x1_5"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
62 "shufflenet_v2_x2_0": {"type": "shufflenet_v2", "model_variant": "x2_0"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
63 "squeezenet1_0": {"type": "squeezenet", "model_variant": "1_0"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
64 "squeezenet1_1": {"type": "squeezenet", "model_variant": "1_1"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
65 "swin_t": {"type": "swin_transformer", "model_variant": "t"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
66 "swin_s": {"type": "swin_transformer", "model_variant": "s"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
67 "swin_b": {"type": "swin_transformer", "model_variant": "b"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
68 "swin_v2_t": {"type": "swin_transformer", "model_variant": "v2_t"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
69 "swin_v2_s": {"type": "swin_transformer", "model_variant": "v2_s"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
70 "swin_v2_b": {"type": "swin_transformer", "model_variant": "v2_b"}, |
3
2c3a3dfaf1a9
planemo upload for repository https://github.com/goeckslab/gleam.git commit 5ab02b6688c9af85525dd225134db8f2bbed76ed
goeckslab
parents:
2
diff
changeset
|
71 "vit_b_16": {"type": "vit", "model_variant": "b_16"}, |
2c3a3dfaf1a9
planemo upload for repository https://github.com/goeckslab/gleam.git commit 5ab02b6688c9af85525dd225134db8f2bbed76ed
goeckslab
parents:
2
diff
changeset
|
72 "vit_b_32": {"type": "vit", "model_variant": "b_32"}, |
2c3a3dfaf1a9
planemo upload for repository https://github.com/goeckslab/gleam.git commit 5ab02b6688c9af85525dd225134db8f2bbed76ed
goeckslab
parents:
2
diff
changeset
|
73 "vit_l_16": {"type": "vit", "model_variant": "l_16"}, |
2c3a3dfaf1a9
planemo upload for repository https://github.com/goeckslab/gleam.git commit 5ab02b6688c9af85525dd225134db8f2bbed76ed
goeckslab
parents:
2
diff
changeset
|
74 "vit_l_32": {"type": "vit", "model_variant": "l_32"}, |
2c3a3dfaf1a9
planemo upload for repository https://github.com/goeckslab/gleam.git commit 5ab02b6688c9af85525dd225134db8f2bbed76ed
goeckslab
parents:
2
diff
changeset
|
75 "vit_h_14": {"type": "vit", "model_variant": "h_14"}, |
2
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
76 "convnext_tiny": {"type": "convnext", "model_variant": "tiny"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
77 "convnext_small": {"type": "convnext", "model_variant": "small"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
78 "convnext_base": {"type": "convnext", "model_variant": "base"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
79 "convnext_large": {"type": "convnext", "model_variant": "large"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
80 "maxvit_t": {"type": "maxvit", "model_variant": "t"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
81 "alexnet": {"type": "alexnet"}, |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
82 "googlenet": {"type": "googlenet"}, |
3
2c3a3dfaf1a9
planemo upload for repository https://github.com/goeckslab/gleam.git commit 5ab02b6688c9af85525dd225134db8f2bbed76ed
goeckslab
parents:
2
diff
changeset
|
83 "inception_v3": {"type": "inceptionv3"}, |
2
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
84 "mobilenet_v2": {"type": "mobilenet_v2"}, |
3
2c3a3dfaf1a9
planemo upload for repository https://github.com/goeckslab/gleam.git commit 5ab02b6688c9af85525dd225134db8f2bbed76ed
goeckslab
parents:
2
diff
changeset
|
85 "mobilenet_v3_large": {"type": "mobilenetv3", "model_variant": "large"}, |
2c3a3dfaf1a9
planemo upload for repository https://github.com/goeckslab/gleam.git commit 5ab02b6688c9af85525dd225134db8f2bbed76ed
goeckslab
parents:
2
diff
changeset
|
86 "mobilenet_v3_small": {"type": "mobilenetv3", "model_variant": "small"}, |
2
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
87 } |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
88 METRIC_DISPLAY_NAMES = { |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
89 "accuracy": "Accuracy", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
90 "accuracy_micro": "Accuracy-Micro", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
91 "loss": "Loss", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
92 "roc_auc": "ROC-AUC", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
93 "roc_auc_macro": "ROC-AUC-Macro", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
94 "roc_auc_micro": "ROC-AUC-Micro", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
95 "hits_at_k": "Hits at K", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
96 "precision": "Precision", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
97 "recall": "Recall", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
98 "specificity": "Specificity", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
99 "kappa_score": "Cohen's Kappa", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
100 "token_accuracy": "Token Accuracy", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
101 "avg_precision_macro": "Precision-Macro", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
102 "avg_recall_macro": "Recall-Macro", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
103 "avg_f1_score_macro": "F1-score-Macro", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
104 "avg_precision_micro": "Precision-Micro", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
105 "avg_recall_micro": "Recall-Micro", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
106 "avg_f1_score_micro": "F1-score-Micro", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
107 "avg_precision_weighted": "Precision-Weighted", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
108 "avg_recall_weighted": "Recall-Weighted", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
109 "avg_f1_score_weighted": "F1-score-Weighted", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
110 "average_precision_macro": "Precision-Average-Macro", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
111 "average_precision_micro": "Precision-Average-Micro", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
112 "average_precision_samples": "Precision-Average-Samples", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
113 "mean_squared_error": "Mean Squared Error", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
114 "mean_absolute_error": "Mean Absolute Error", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
115 "r2": "R² Score", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
116 "root_mean_squared_error": "Root Mean Squared Error", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
117 "mean_absolute_percentage_error": "Mean Absolute % Error", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
118 "root_mean_squared_percentage_error": "Root Mean Squared % Error", |
186424a7eca7
planemo upload for repository https://github.com/goeckslab/gleam.git commit 91fa4aba245520fc0680088a07cead66bcfd4ed2
goeckslab
parents:
diff
changeset
|
119 } |