YAML Metadata Warning:empty or missing yaml metadata in repo card

Check out the documentation for more information.

''' 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 ''''

Downloads last month: 1

Inference Providers NEW

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support