[go: nahoru, domu]

Skip to content

Crawl data, process data, visualize, and create ML model for laptop price prediction

License

Notifications You must be signed in to change notification settings

hellofromtheothersky/Laptop-price-analysis-and-prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Laptop price analysis and prediction

This project is finished with the help of my collaborators: Trinh The Hien supporting the missing data handling task and prediction model, Luu Quang Tien Hoang and Nguyen Thanh Nhan who make this project more complete.

--

Problem description:

The project goal is to predict the price of a laptop based on its configuration

  • Input: 'weight', 'Processor rank', 'Graphics Coprocessor perf', 'Brand', 'Laptop type', 'Laptop purpose', 'Hard Drive Type', 'Memory Type', and 'Operating System'
  • Output: Price

Although there are above 70 features of the laptop from the raw data, after many phases of cleaning data, almost all of them just have one value, we choose those features above which have the least null value and the most affected to our target label

Data collecting

The data is collected from 3 different sources:

  • cpubenchmark.net
  • notebookcheck.net/Mobile-Graphics-Cards-Benchmark-List.844.0.html
  • amazon.com

*Detail about how I crawl data from amazon is currently in my another repository

Amazon website provides laptops and laptop configuration data

The first two websites provide more details about the CPU/GPU name and its ranking/performance data.

So while I need this, to predict the price of a laptop product, CPU and GPU are definitely such valuable features. But just relying on the amazon data, we only have the string name of it. It means to make the ML model learn these features, we can only use One-hot-encoding, but CPU and GPU have 503 and 191 unique values, respectively. And to both reduce the model complexity and increase the data understanding ability, I extra collect another data source, and then map the string name to an integer value (ranking/performance)

Data preprocessing

Here are some key steps in this phase:

  • Map the string name of CPU/GPU to their Rank/Performance as an integer: Using rule-based and Levenshtein distance to match the raw name and its standard name: 'intel i5 1135 g7' -> 'intel core i5-113g7' -> RANK NUMBER
  • Some single features have more than one kind of information will be split: "4 GB DDR4" into "4 GB" for "Memory size" and "DDR4" for "Memory type"
  • Transform and remove value having measurement symbol

EDA

I did some Exploratory data analysis that can be founded in the notebook/EDA.ipynb

Model

Our best-performed model is Random Forest Regressor with R2 score is approximately 0.75

How to run

  • Copy 2 files "laptop_list.csv" and "laptop_details.json" which is the output of the crawler (details here) to the data/raw/ folder
  • Run the 'src/preprocessing/cleaning_data.py' file and then 'src/preprocessing/deal_missing_val.py', it will generate final_data.csv in the data/processed
  • Run this command python3 src/modeling/train.py to train the model (default is Random Forest), or you can choose the model via --mdname: python3 src/modeling/train.py --mdname dc for decision tree

About

Crawl data, process data, visualize, and create ML model for laptop price prediction

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published