Tomertg commited on
Commit
a30c2ac
·
verified ·
1 Parent(s): 88d98af

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +34 -0
README.md CHANGED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # 🏋️‍♂️ Gradient Boosting Deadlift Predictor
2
+
3
+ This repository contains the winning model from Assignment #2: Classification, Regression, Clustering & Evaluation.
4
+
5
+ ## 📌 Model Purpose
6
+ The model predicts an athlete's **deadlift performance (lbs)** based on physical and strength-related features.
7
+
8
+ ## 🧠 Algorithm
9
+ ✅ Gradient Boosting Regressor
10
+ Selected as the final model after comparing:
11
+
12
+ - Linear Regression
13
+ - Random Forest
14
+ - Gradient Boosting
15
+
16
+ ## 🏆 Performance (Test Set)
17
+
18
+ - R²: 0.85
19
+ - MAE: ~28.6 lbs
20
+ - RMSE: ~37.2 lbs
21
+
22
+ Gradient Boosting achieved the **highest accuracy and lowest error**, so it was chosen as the final model.
23
+
24
+ ## 📁 Files
25
+ - `winning_model.pkl` – serialized model ready for loading and inference
26
+
27
+ ## 🔧 Usage
28
+ ```python
29
+ import pickle
30
+
31
+ with open("winning_model.pkl", "rb") as f:
32
+ model = pickle.load(f)
33
+
34
+ prediction = model.predict([[weight, height, backsquat, snatch]])