- python3
- keras >= 2.1.5
- sklearn >= 0.0
src/predict.py
, builds predicting/embedding modelsrc/analysis.py
, uses TSNE and KMeans to classify the moviesMovie-Rating-Problem-鄭雅文.pdf
, the reportfigures/
, the figures of the reportoutput data/
, the csv files for analysis
python3 src/predict.py [rating file]
It will generate a file named 'testing.h5' (keras model) and a file named 'predict.csv' (predicted rating).
python3 src/analysis.py
It will do classification.
Jeng, Ya-wen (鄭雅文) r06922097@ntu.edu.tw
- MovieLens 20M Dataset, movie dataset
- Plotly, graph maker