Code for training and evaluating an Airbnb price prediction model. Made by Team Data Viz-ards for CSE 6242 Project, Spring 2020.
Ensure that you have Python 3.7.6 and virtualenv
installed. Then perform the following steps:
- Clone or download this repository and
cd
into it. - Create a Python 3.7.6
virtualenv
, and activate it:
virtualenv --python=path_to_your_python_executable venv
source venv/bin/activate
- Install dependencies using
pip
:
pip install -r requirements.txt
Simply run:
python AirbnbModel.py
This will train and evaluate the model on the provided dataset - updatedFinalData.csv
and generate output files to store predictions, feature importance values, and save the model to disk.