Downloads · 30 days
16
28% of all-time downloads
mentallongformer-cams-finetuned
stackofsugar/mentallongformer-cams-finetuned
mentallongformer-cams-finetuned is a text classification model from stackofsugar. Use it when you need a label for a piece of text. It is set up for transformers. The card lists the license as mit.
This model is fine-tuned from the checkpoint of AIMH/mental-longformer-base-4096 using drmuskangarg/CAMS dataset. For more information about the base Longformer model, please visit their model page. We used the same c…
All-time downloads
58
Public
Parameters
149M
1.2 GB on disk
Likes
2
Public
Repo makeup
Click a slice to open those files.
.safetensors595 MB · 100%
At a glance
- Task
- Text Classification
- Library
- transformers
- License
- mit
- Model type
- longformer
- Access
- Public
- Created
- Oct 30, 2024
- Updated
- Nov 1, 2024
- SHA
- 9ca38d8a
Try a prompt
Base models
Model card
From the Hugging Face model README
About This Model
This model is fine-tuned from the checkpoint of AIMH/mental-longformer-base-4096 using drmuskangarg/CAMS dataset. For more information about the base Longformer model, please visit their model page. We used the same configuration as AIMH/mental-longformer-base-4096 including their tokenizer.
Usage
If you wish to use my model to infer your dataset or maybe pre-train it further, you can import my model in a Python script/notebook.
from transformers import LongformerTokenizer, LongformerForSequenceClassification
tokenizer = LongformerTokenizer.from_pretrained("aimh/mental-longformer-base-4096")
model = LongformerForSequenceClassification.from_pretrained("stackofsugar/mentallongformer-cams-finetuned")
If you prefer to use the high-level HuggingFace pipeline to make predictions, you can also do it in a Python script/notebook.
from transformers import pipeline
pipe = pipeline("text-classification", model="stackofsugar/mentallongformer-cams-finetuned", tokenizer="aimh/mental-longformer-base-4096")
More Information
For more information, visit my GitHub Repo.
- Task
- Text Classification
- Library
- transformers
- Type
- longformer
- License
- mit
- Languages
- en
- Created
- Oct 30, 2024
- Updated
- Nov 1, 2024