File size: 806 Bytes
798791f
e043b10
 
 
798791f
 
e043b10
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
---
base_model: InstaDeepAI/nucleotide-transformer-v2-100m-multi-species
tags:
- generated_from_trainer
metrics:
- f1
model-index:
- name: vedantM/NucArg_LongRead
  results: []
---

# NucArg: Antibiotic Resistance Detection

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.
It achieves the following results on the evaluation set:
- Loss: 0.467312
- F1: 0.891351
- Recall: 0.891115
- Precision: 0.893135

## Training hyperparameters

The following hyperparameters were used during training:
- learning_rate: 0.000025
- train_batch_size: 8
- gradient_accumulation_steps= 8
- eval_batch_size: 64
- seed: 42
- lr_scheduler_type: cosine
- training_epoch: 5