Skip to content

Just use raw data, predict the actual rating of a user give to a specific movie

Notifications You must be signed in to change notification settings

zhazhangqi/ANN-KNN_Prediction_Model

 
 

Repository files navigation

ANN_Prediction_Model

Just use raw data, predict the actual rating of a user give to a specific movie. The main folder contains the raw data and data preprocessing code, which is written in R.

##sampleData This folder contains the sample training and testing data. Data preprocess code are written in R.

###readRawData.R Preprocess the raw data, cluster the movie into 1000 clusters according to its name. The distance between two strings is calculated by Jaro–Winkler distance.

###movie_avg.csv This file contains the average rating score of each movie, all the data are computed via the trainDataset.

##ANN_prediction Build an artificial neural network(ANN) classification model using 100,000 sample data from the raw data. Calculate the RMSE, error rate of the model. Neural Network Toolbox license is needed.

##KNN_prediction Build an k-nearest neighbors(KNN) classification model using 100,000 sample data from the raw data. Calculate the RMSE, error rate of the model.

About

Just use raw data, predict the actual rating of a user give to a specific movie

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • MATLAB 58.8%
  • R 41.2%