OloriBern's picture
Upload README.md with huggingface_hub
8e25157 verified
|
raw
history blame
1.01 kB
---
library_name: sklearn
tags:
  - sklearn
  - regression
  - embeddings
  - weight-prediction
  - elastic
model-index:
  - name: Elastic Net
    results:
      - task:
          type: regression
          name: Embedding Weight Prediction
        metrics:
          - type: mse
            value: 0.7620290676891099
            name: Test MSE
          - type: r2
            value: -0.044365932705230184
            name: Test R²
---
# Elastic Net Weight Predictor

Linear model combining L1 and L2 regularization

## Performance Metrics

- Training Time: 1.65 seconds
- Training MSE: 0.739313
- Testing MSE: 0.762029
- Training R²: 0.003906
- Testing R²: -0.044366

## Usage

```python
import skops.io as sio

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

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