| { | |
| "architectures": ["EfficientNet"], | |
| "model_type": "efficientnet_b0", | |
| "num_classes": 33, | |
| "image_size": 224, | |
| "num_channels": 3, | |
| "mean": [0.485, 0.456, 0.406], | |
| "std": [0.229, 0.224, 0.225], | |
| "framework": "pytorch", | |
| "library": "timm", | |
| "task": "image-classification", | |
| "dataset": "DIBaS", | |
| "id2label": { | |
| "0": "Acinetobacter_baumannii", | |
| "1": "Actinomyces_israelii", | |
| "2": "Bacteroides_fragilis", | |
| "3": "Bifidobacterium_spp", | |
| "4": "Candida_albicans", | |
| "5": "Clostridium_perfringens", | |
| "6": "Enterococcus_faecalis", | |
| "7": "Enterococcus_faecium", | |
| "8": "Escherichia_coli", | |
| "9": "Fusobacterium", | |
| "10": "Lactobacillus_casei", | |
| "11": "Lactobacillus_crispatus", | |
| "12": "Lactobacillus_delbrueckii", | |
| "13": "Lactobacillus_gasseri", | |
| "14": "Lactobacillus_jensenii", | |
| "15": "Lactobacillus_johnsonii", | |
| "16": "Lactobacillus_paracasei", | |
| "17": "Lactobacillus_plantarum", | |
| "18": "Lactobacillus_reuteri", | |
| "19": "Lactobacillus_rhamnosus", | |
| "20": "Lactobacillus_salivarius", | |
| "21": "Listeria_monocytogenes", | |
| "22": "Micrococcus_spp", | |
| "23": "Neisseria_gonorrhoeae", | |
| "24": "Porphyromonas_gingivalis", | |
| "25": "Propionibacterium_acnes", | |
| "26": "Proteus", | |
| "27": "Pseudomonas_aeruginosa", | |
| "28": "Staphylococcus_aureus", | |
| "29": "Staphylococcus_epidermidis", | |
| "30": "Staphylococcus_saprophyticus", | |
| "31": "Streptococcus_agalactiae", | |
| "32": "Veillonella" | |
| }, | |
| "label2id": { | |
| "Acinetobacter_baumannii": 0, | |
| "Actinomyces_israelii": 1, | |
| "Bacteroides_fragilis": 2, | |
| "Bifidobacterium_spp": 3, | |
| "Candida_albicans": 4, | |
| "Clostridium_perfringens": 5, | |
| "Enterococcus_faecalis": 6, | |
| "Enterococcus_faecium": 7, | |
| "Escherichia_coli": 8, | |
| "Fusobacterium": 9, | |
| "Lactobacillus_casei": 10, | |
| "Lactobacillus_crispatus": 11, | |
| "Lactobacillus_delbrueckii": 12, | |
| "Lactobacillus_gasseri": 13, | |
| "Lactobacillus_jensenii": 14, | |
| "Lactobacillus_johnsonii": 15, | |
| "Lactobacillus_paracasei": 16, | |
| "Lactobacillus_plantarum": 17, | |
| "Lactobacillus_reuteri": 18, | |
| "Lactobacillus_rhamnosus": 19, | |
| "Lactobacillus_salivarius": 20, | |
| "Listeria_monocytogenes": 21, | |
| "Micrococcus_spp": 22, | |
| "Neisseria_gonorrhoeae": 23, | |
| "Porphyromonas_gingivalis": 24, | |
| "Propionibacterium_acnes": 25, | |
| "Proteus": 26, | |
| "Pseudomonas_aeruginosa": 27, | |
| "Staphylococcus_aureus": 28, | |
| "Staphylococcus_epidermidis": 29, | |
| "Staphylococcus_saprophyticus": 30, | |
| "Streptococcus_agalactiae": 31, | |
| "Veillonella": 32 | |
| }, | |
| "metrics": { | |
| "validation_accuracy": 0.9167, | |
| "macro_f1": 0.917, | |
| "parameters": 4050977 | |
| }, | |
| "training": { | |
| "epochs": 10, | |
| "batch_size": 32, | |
| "optimizer": "AdamW", | |
| "learning_rate": 0.001, | |
| "weight_decay": 0.0001, | |
| "mixed_precision": true, | |
| "seed": 42 | |
| } | |
| } | |