Downloads · 30 days
29
3% of all-time downloads
Albert-IT-JobRecommendation
Apizhai/Albert-IT-JobRecommendation
Albert-IT-JobRecommendation is a text classification model from Apizhai. Use it when you need a label for a piece of text. It is set up for transformers.
This repository contains a Albert model designed for text classification. The architecture of the model is based on the Albert Base v2 model.
All-time downloads
833
Public
Repo size
94.3 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin46.8 MB · 98%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- albert
- Access
- Public
- Created
- Dec 24, 2023
- Updated
- Mar 11, 2024
- SHA
- fd28763a
Try a prompt
Model card
From the Hugging Face model README
This repository contains a Albert model designed for text classification. The architecture of the model is based on the Albert Base v2 model.
Library
pip install transformers
pip install sentencepiece
Example
from transformers import AutoModel,AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained('Apizhai/Albert-IT-JobRecommendation', use_fast=False),
model = AutoModel.from_pretrained('Apizhai/Albert-IT-JobRecommendation')
Training hyperparameters
The following hyperparameters were used during training:
- max_seq_length: 128
- max_length: 128
- train_batch_size: 4
- eval_batch_size: 32
- num_train_epochs: 10
- evaluate_during_training: False
- evaluate_during_training_steps: 100
- use_multiprocessing: False
- fp16: True
- save_steps: -1
- save_eval_checkpoints: False
- save_model_every_epoch: False
- no_cache: True
- reprocess_input_data: True
- overwrite_output_dir: True
- preprocess_inputs: False
- num_return_sequences: 1
Score
- f1-score: 0.85574
- macro avg: 0.84748
- weighted avg: 0.81575
- Task
- Text Classification
- Library
- transformers
- Type
- albert
- Languages
- en
- Created
- Dec 24, 2023
- Updated
- Mar 11, 2024