This part of the project handles data preprocessing, model training, recommendation generation, and evaluation.
The Python part of the project reads movie and ratings data, preprocesses it, splits it into training and test sets, trains a collaborative filtering model, generates recommendations, and evaluates the model's performance.
python/
├── Dockerfile
├── requirements.txt
├── preprocess.py
├── main.py
├── database.py
├── similarity.py
├── recommendations.py
└── tests/
├── test_database.py
├── test_integration.py
├── test_preprocess.py
├── test_recommendations.py
└── test_similarity.py
- Python 3.x
- Install dependencies using
requirements.txt:
pip install -r requirements.txtRun the preprocessing script to preprocess the data and split it into training and test sets:
python preprocess.pyRun the main script to train the model, generate recommendations, and evaluate the model:
python main.pypytest -s testsThis project is licensed under the MIT License.