This project focuses on predicting the price range of mobile phones based on their specifications. We experimented with different machine learning models to compare their performance and find the most accurate one.
Dataset: Mobile Price Classification (Kaggle)
battery_power: Total energy a battery can store in one time measured in mAhblue: Has bluetooth or notclock_speed: speed at which microprocessor executes instructionsdual_sim: Has dual sim support or notfc: Front Camera mega pixelsfour_g: Has 4G or notint_memory: Internal Memory in Gigabytesm_dep: Mobile Depth in cmn_cores: Number of cores of processorpc: Primary Camera mega pixelspx_height: Pixel Resolution Heightram: Random Access Memory in Megabytessc_h: Screen Height of mobile in cmsc_w: Screen Width of mobile in cmthree_g: Has 3G or nottouch_screen: Has touch screen or notwifi: Has wifi or nottalk_time: longest time that a single battery charge will last when you areprice_range(Target): This is the target variable with value of 0(low cost), 1(medium cost), 2(high cost) and 3(very high cost).
- Features with high correlation to the target variable were selected to train the models.
- The Features is :
battery_power,px_height,px_width,ram - This ensures better accuracy and reduces noise from irrelevant features.
We trained and tested four ML models:
| Model | Accuracy (%) |
|---|---|
| Decision Tree | 86.5 |
| K-Nearest Neighbors | 93.0 |
| Support Vector Machine | 96.75 |
| Logistic Regression | 97.25 |
β Logistic Regression achieved the highest accuracy (97.25%), making it the best-performing model in this project.