ihoflaz commited on
Commit
bf1065e
·
verified ·
1 Parent(s): 74dae58

Upload config.json with huggingface_hub

Browse files
Files changed (1) hide show
  1. config.json +97 -0
config.json ADDED
@@ -0,0 +1,97 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": ["ResNet50"],
3
+ "model_type": "resnet50",
4
+ "num_classes": 33,
5
+ "image_size": 224,
6
+ "num_channels": 3,
7
+ "mean": [0.485, 0.456, 0.406],
8
+ "std": [0.229, 0.224, 0.225],
9
+ "framework": "pytorch",
10
+ "library": "timm",
11
+ "task": "image-classification",
12
+ "dataset": "DIBaS",
13
+ "id2label": {
14
+ "0": "Acinetobacter_baumannii",
15
+ "1": "Actinomyces_israelii",
16
+ "2": "Bacteroides_fragilis",
17
+ "3": "Bifidobacterium_spp",
18
+ "4": "Candida_albicans",
19
+ "5": "Clostridium_perfringens",
20
+ "6": "Enterococcus_faecalis",
21
+ "7": "Enterococcus_faecium",
22
+ "8": "Escherichia_coli",
23
+ "9": "Fusobacterium",
24
+ "10": "Lactobacillus_casei",
25
+ "11": "Lactobacillus_crispatus",
26
+ "12": "Lactobacillus_delbrueckii",
27
+ "13": "Lactobacillus_gasseri",
28
+ "14": "Lactobacillus_jensenii",
29
+ "15": "Lactobacillus_johnsonii",
30
+ "16": "Lactobacillus_paracasei",
31
+ "17": "Lactobacillus_plantarum",
32
+ "18": "Lactobacillus_reuteri",
33
+ "19": "Lactobacillus_rhamnosus",
34
+ "20": "Lactobacillus_salivarius",
35
+ "21": "Listeria_monocytogenes",
36
+ "22": "Micrococcus_spp",
37
+ "23": "Neisseria_gonorrhoeae",
38
+ "24": "Porphyromonas_gingivalis",
39
+ "25": "Propionibacterium_acnes",
40
+ "26": "Proteus",
41
+ "27": "Pseudomonas_aeruginosa",
42
+ "28": "Staphylococcus_aureus",
43
+ "29": "Staphylococcus_epidermidis",
44
+ "30": "Staphylococcus_saprophyticus",
45
+ "31": "Streptococcus_agalactiae",
46
+ "32": "Veillonella"
47
+ },
48
+ "label2id": {
49
+ "Acinetobacter_baumannii": 0,
50
+ "Actinomyces_israelii": 1,
51
+ "Bacteroides_fragilis": 2,
52
+ "Bifidobacterium_spp": 3,
53
+ "Candida_albicans": 4,
54
+ "Clostridium_perfringens": 5,
55
+ "Enterococcus_faecalis": 6,
56
+ "Enterococcus_faecium": 7,
57
+ "Escherichia_coli": 8,
58
+ "Fusobacterium": 9,
59
+ "Lactobacillus_casei": 10,
60
+ "Lactobacillus_crispatus": 11,
61
+ "Lactobacillus_delbrueckii": 12,
62
+ "Lactobacillus_gasseri": 13,
63
+ "Lactobacillus_jensenii": 14,
64
+ "Lactobacillus_johnsonii": 15,
65
+ "Lactobacillus_paracasei": 16,
66
+ "Lactobacillus_plantarum": 17,
67
+ "Lactobacillus_reuteri": 18,
68
+ "Lactobacillus_rhamnosus": 19,
69
+ "Lactobacillus_salivarius": 20,
70
+ "Listeria_monocytogenes": 21,
71
+ "Micrococcus_spp": 22,
72
+ "Neisseria_gonorrhoeae": 23,
73
+ "Porphyromonas_gingivalis": 24,
74
+ "Propionibacterium_acnes": 25,
75
+ "Proteus": 26,
76
+ "Pseudomonas_aeruginosa": 27,
77
+ "Staphylococcus_aureus": 28,
78
+ "Staphylococcus_epidermidis": 29,
79
+ "Staphylococcus_saprophyticus": 30,
80
+ "Streptococcus_agalactiae": 31,
81
+ "Veillonella": 32
82
+ },
83
+ "metrics": {
84
+ "validation_accuracy": 0.9394,
85
+ "macro_f1": 0.939,
86
+ "parameters": 23575649
87
+ },
88
+ "training": {
89
+ "epochs": 20,
90
+ "batch_size": 24,
91
+ "optimizer": "AdamW",
92
+ "learning_rate": 0.001,
93
+ "weight_decay": 0.0001,
94
+ "mixed_precision": true,
95
+ "seed": 42
96
+ }
97
+ }