vedantM commited on
Commit
2ec03d4
·
verified ·
1 Parent(s): 3a484d9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +28 -8
README.md CHANGED
@@ -1,11 +1,31 @@
1
  ---
2
- datasets:
3
- - vedantM/NucArg
 
4
  metrics:
5
  - f1
6
- - accuracy
7
- - precision
8
- - recall
9
- base_model:
10
- - InstaDeepAI/nucleotide-transformer-v2-100m-multi-species
11
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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_ShortRead
9
+ results: []
10
+ ---
11
+
12
+ # nucleotide-transformer-finetuned-lora-NucleotideTransformer
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.675450
17
+ - F1: 0.823285
18
+ - Recall: 0.824769
19
+ - Precision: 0.822969
20
+
21
+ ## Training hyperparameters
22
+
23
+ The following hyperparameters were used during training:
24
+ - learning_rate: 0.000025
25
+ - train_batch_size: 32
26
+ - gradient_accumulation_steps= 8
27
+ - eval_batch_size: 32
28
+ - seed: 42
29
+ - lr_scheduler_type: linear
30
+ - training_epoch: 4
31
+