/amazon-books-4-class-rating-prediction

This repository holds the dataset and notebooks for the Amazon Books dataset 4 class Rating prediction

Primary LanguageJupyter Notebook

amazon-books-4-class-rating-prediction

This repository holds the dataset and notebooks for the Amazon Books dataset 4 class Rating prediction

Dataset Used

Dataset link : Amazon Books Dataset

The dataset contains four ratings: 2, 3, 4, 5

Feature Extraction Technique

  • Term Frequency - Inverse Document Frequency

Resampling Technique Used

  • Random Oversampling

Machine Learning Algorithms Used

  • K Nearest Neighbors
  • Decision Tree
  • Support Vector Machine
  • Multinomial Naive-Bayes
  • Random Forest