| --- | |
| 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) | |
| ``` | |