YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/model-cards#model-card-metadata)
---
library_name: sklearn
tags:
  - sklearn
  - regression
  - embeddings
  - weight-prediction
  - ridge
model-index:
  - name: Ridge Regression
    results:
      - task:
          type: regression
          name: Embedding Weight Prediction
        metrics:
          - type: mse
            value: 0.752126161053798
            name: Test MSE
          - type: r2
            value: -0.03790253867179685
            name: Test R²
---
# Ridge Regression Weight Predictor

Linear model with L2 regularization for handling multicollinearity

## Performance Metrics

- Training Time: 15.79 seconds
- Training MSE: 0.460125
- Testing MSE: 0.752126
- Training R²: 0.384710
- Testing R²: -0.037903

## Model Analysis

### Predictions vs True Values
![Predictions](./plots/predictions.png)

This plot shows how well the model's predictions match the true values:
- Points on the red line indicate perfect predictions
- Spread around the line shows prediction uncertainty
- Systematic deviations indicate bias

### Error Distribution
![Error Distribution](./plots/error_distribution.png)

This plot shows the distribution of prediction errors:
- Centered around zero indicates unbiased predictions
- Width shows prediction precision
- Shape reveals error patterns

### Dimension-wise Performance
![Dimension MSE](./plots/dimension_mse.png)

This plot shows the MSE for each embedding dimension:
- Lower bars indicate better predictions
- Variations show which dimensions are harder to predict
- Can guide targeted improvements

## Usage

```python
import skops.io as sio

# Load the model
model = sio.load('weight_predictor_ridge.skops')

# Make predictions
weights = model.predict(question_embedding)
```
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The model has no library tag.