OloriBern commited on
Commit
c0da5f4
·
verified ·
1 Parent(s): a0af860

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +47 -0
README.md ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ library_name: sklearn
4
+ tags:
5
+ - sklearn
6
+ - regression
7
+ - embeddings
8
+ - weight-prediction
9
+ - ridge
10
+ model-index:
11
+ - name: Ridge Regression
12
+ results:
13
+ - task:
14
+ type: regression
15
+ name: Embedding Weight Prediction
16
+ metrics:
17
+ - type: mse
18
+ value: 0.752126161053798
19
+ name: Test MSE
20
+ - type: r2
21
+ value: -0.03790253867179685
22
+ name: Test R²
23
+ ---
24
+ # Ridge Regression Weight Predictor
25
+
26
+ Linear model with L2 regularization for handling multicollinearity
27
+
28
+ ## Performance Metrics
29
+
30
+ - Training Time: 16.09 seconds
31
+ - Training MSE: 0.460125
32
+ - Testing MSE: 0.752126
33
+ - Training R²: 0.384710
34
+ - Testing R²: -0.037903
35
+
36
+ ## Usage
37
+
38
+ ```python
39
+ import skops.io as sio
40
+
41
+ # Load the model
42
+ model = sio.load('weight_predictor_ridge.skops')
43
+
44
+ # Make predictions
45
+ weights = model.predict(question_embedding)
46
+ ```
47
+