vedantM commited on
Commit
e043b10
·
verified ·
1 Parent(s): 798791f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +28 -13
README.md CHANGED
@@ -1,16 +1,31 @@
1
  ---
2
- datasets:
3
- - vedantM/NucArg
 
4
  metrics:
5
  - f1
6
- - accuracy
7
- - recall
8
- - precision
9
- base_model:
10
- - InstaDeepAI/nucleotide-transformer-v2-100m-multi-species
11
- tags:
12
- - biology
13
- - genomics
14
- - DNA
15
- - Antibiotic
16
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ base_model: InstaDeepAI/nucleotide-transformer-v2-100m-multi-species
3
+ tags:
4
+ - generated_from_trainer
5
  metrics:
6
  - f1
7
+ model-index:
8
+ - name: vedantM/NucArg_LongRead
9
+ results: []
10
+ ---
11
+
12
+ # NucArg: Antibiotic Resistance Detection
13
+
14
+ This model is a fine-tuned version of [InstaDeepAI/nucleotide-transformer-v2-100m-multi-species](https://huggingface.co/InstaDeepAI/nucleotide-transformer-v2-100m-multi-species) on the NucArg dataset.
15
+ It achieves the following results on the evaluation set:
16
+ - Loss: 0.467312
17
+ - F1: 0.891351
18
+ - Recall: 0.891115
19
+ - Precision: 0.893135
20
+
21
+ ## Training hyperparameters
22
+
23
+ The following hyperparameters were used during training:
24
+ - learning_rate: 0.000025
25
+ - train_batch_size: 8
26
+ - gradient_accumulation_steps= 8
27
+ - eval_batch_size: 64
28
+ - seed: 42
29
+ - lr_scheduler_type: cosine
30
+ - training_epoch: 5
31
+