commited on
Commit
ce56872
·
verified ·
1 Parent(s): b0ac4db

Upload 3 files

Browse files
Files changed (3) hide show
  1. README.md +60 -0
  2. model_columns.pkl +3 -0
  3. rf_model.pkl +3 -0
README.md CHANGED
@@ -1,3 +1,63 @@
1
  ---
 
 
 
 
 
 
 
2
  license: mit
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ title: "💎 Gemstone Price Regression"
3
+ emoji: 💰
4
+ colorFrom: indigo
5
+ colorTo: blue
6
+ sdk: streamlit
7
+ app_file: app.py
8
+ pinned: true
9
  license: mit
10
+ tags:
11
+ - regression
12
+ - machine-learning
13
+ - streamlit
14
+ - diamonds
15
+ - synthetic-data
16
  ---
17
+
18
+ # 💎 Gemstone Price Prediction App
19
+
20
+ This Streamlit app predicts the price of a gemstone using its physical and quality-related features.
21
+
22
+ ## 🧠 Project Overview
23
+
24
+ - This project simulates a **gemstone pricing system** using synthetic tabular data.
25
+ - Features include: `carat`, `depth`, `table`, `x`, `y`, `z`, `clarity_score`, `color_score`, and `cut_score`.
26
+ - The target variable is **price** (USD).
27
+ - Model: **RandomForestRegressor**
28
+ - Trained on 1000 synthetic samples.
29
+
30
+ ---
31
+
32
+ ## 📊 Performance
33
+
34
+ - RMSE: **605.16**
35
+ - R² Score: **0.9549**
36
+
37
+ ---
38
+
39
+ ## 🚀 How to Run Locally
40
+
41
+ ```bash
42
+ pip install -r requirements.txt
43
+ streamlit run app.py
44
+
45
+
46
+
47
+ 🔮 Future Work
48
+ Area Improvement
49
+ Model Try XGBoost, LightGBM
50
+ Feature Engineering Interaction terms, log/carat scaling
51
+ Deployment Add API endpoint with FastAPI
52
+ Real-world Data Integrate real gemstone datasets
53
+
54
+
55
+ 📁 Files
56
+ app.py: Streamlit interface
57
+
58
+ rf_model.pkl: Trained model
59
+
60
+ model_columns.pkl: List of input features
61
+
62
+ requirements.txt: Required libraries
63
+
model_columns.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5ba62c0315c390a17bf49a71aa039e95ff212a5951b61580313c9e309f56c9c8
3
+ size 94
rf_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7e41679192d03f13b8bbfc23b104659d12bc2b9966bc5126a54e30cbdcf36328
3
+ size 7300001