Downloads · 30 days
5
38% of all-time downloads
bert-personality-big5_nasser_big_data
margati/bert-personality-big5_nasser_big_data
bert-personality-big5_nasser_big_data is a machine learning model from margati. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
''' MSE per trait: extraversion: 0.0203 neuroticism: 0.0205 agreeableness: 0.0165 conscientiousness: 0.0202 openness: 0.0190 '''
All-time downloads
13
Public
Repo size
267 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin134 MB · 99%
At a glance
- Model type
- bert
- Access
- Public
- Created
- Jul 9, 2025
- Updated
- Jul 10, 2025
- SHA
- cfc9c32a
Model card
From the Hugging Face model README
''' MSE per trait: extraversion: 0.0203 neuroticism: 0.0205 agreeableness: 0.0165 conscientiousness: 0.0202 openness: 0.0190 '''
num_train_epochs=3,
per_device_train_batch_size=4,
per_device_eval_batch_size=4,
warmup_steps=50,
weight_decay=0.01,
learning_rate= 2e-5, #
logging_dir='./logs',
logging_steps=20, #
eval_strategy="steps",
eval_steps=500,#
save_strategy="steps",
save_steps=500,
load_best_model_at_end=True,
metric_for_best_model="mse",
greater_is_better=False,
report_to=None,
dataloader_pin_memory=False,
''' Pearson Correlation per trait: extraversion: 0.3346 neuroticism: 0.3652 agreeableness: 0.2984 conscientiousness: 0.3721 openness: 0.3400 ''''
- Type
- bert
- Created
- Jul 9, 2025
- Updated
- Jul 10, 2025