slm-pricer-slm-pricer-e2e-20260204-1layer-3epochs-regressor-20260206

Regression head trained for price prediction.

Metrics

• val_mae: 40.32
• train_frac_pct: 100

Configuration

• dataset_name: ('ed-donner/items_prompts_full', 'sfmocha/items_prompts_full')
• model_name: antonawinkler/slm-pricer-e2e-20260204-1layer-3epochs
• y_transform: log
• batch_size: 2048
• learning_rate: 0.0001
• weight_decay: 0.05
• dropout: 0.2
• epochs: 100
• patience: 30
• grad_clip: 10.0
• input_dim: 3072

Usage

from slm_pricer.training import load_regression_head_from_hub

model, metadata = load_regression_head_from_hub("antonawinkler/slm-pricer-slm-pricer-e2e-20260204-1layer-3epochs-regressor-20260206")
print(f"Val MAE: ${metadata['metrics']['val_mae']:.2f}")