OloriBern commited on
Commit
8e25157
·
verified ·
1 Parent(s): bbf98a1

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
+ - elastic
10
+ model-index:
11
+ - name: Elastic Net
12
+ results:
13
+ - task:
14
+ type: regression
15
+ name: Embedding Weight Prediction
16
+ metrics:
17
+ - type: mse
18
+ value: 0.7620290676891099
19
+ name: Test MSE
20
+ - type: r2
21
+ value: -0.044365932705230184
22
+ name: Test R²
23
+ ---
24
+ # Elastic Net Weight Predictor
25
+
26
+ Linear model combining L1 and L2 regularization
27
+
28
+ ## Performance Metrics
29
+
30
+ - Training Time: 1.65 seconds
31
+ - Training MSE: 0.739313
32
+ - Testing MSE: 0.762029
33
+ - Training R²: 0.003906
34
+ - Testing R²: -0.044366
35
+
36
+ ## Usage
37
+
38
+ ```python
39
+ import skops.io as sio
40
+
41
+ # Load the model
42
+ model = sio.load('weight_predictor_elastic.skops')
43
+
44
+ # Make predictions
45
+ weights = model.predict(question_embedding)
46
+ ```
47
+