Downloads · 30 days
17
8% of all-time downloads
sasasassaszzd/PHQ8-prototype
PHQ8-prototype is a feature extraction model from sasasassaszzd. Use it when you need embeddings to search or compare text. It is set up for transformers.
This is a custom BERT-based model with an MLP head for PHQ-8 score prediction.
Downloads · 30 days
17
8% of all-time downloads
All-time downloads
213
Public
Parameters
125M
3.6 GB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors501 MB · 99%
From the Hugging Face model README
This is a custom BERT-based model with an MLP head for PHQ-8 score prediction.
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("username/PHQ8-prototype")
model = AutoModel.from_pretrained("username/PHQ8-prototype", trust_remote_code=True)
inputs = tokenizer("I feel tired and down.", return_tensors="pt")
outputs = model.inference(**inputs)
print(outputs)