tydymy's picture
Create README.md
2ef5159
# Model Card for gena_lm_bert_base_human_classify
## Model Details
- **Model Name:** gena_lm_bert_base_human_classify
- **Type:** Transformer
- **Main Application:** [Brief description of the main application of the model, e.g., text classification, image recognition, etc.]
## Training Data
- **Description:** [Brief description of the training data, including source, nature (text, images, etc.), and size.]
- **Preprocessing:** [Details of any preprocessing steps applied to the training data.]
## Model Architecture
- **Architecture Details:** [Details about the model architecture, e.g., number of layers, type of layers, etc.]
- **Framework Used:** PyTorch
## Training Procedure
- **Epochs:** 2
- **Batch Size:** 64
- **Learning Rate:** 2e-5
- **Weight Decay:** 0.01
### Training Performance
- **Epoch 1:**
- Training Loss: 0.088600
- Validation Loss: 0.079851
- Accuracy: 96.9529%
- F1 Score: 96.9840%
- Precision: 95.7133%
- Recall: 98.2889%
- **Epoch 2:**
- Training Loss: 0.058800
- Validation Loss: 0.057442
- Accuracy: 98.0280%
- F1 Score: 98.0338%
- Precision: 97.4486%
- Recall: 98.6260%