search

Jayshah6699 / datascience-mashup

In this repo I will try to gather all of the projects related to data science with clean datasets and high accuracy models to solve real world problems.
MIT License
41 stars 53 forks source link

Added Netflix EDA and Genre Prediction using Bert #303

Closed infiniteoverflow closed 3 years ago

infiniteoverflow commented 3 years ago

Fixes issue #289

Dataset : https://www.kaggle.com/shivamb/netflix-shows
Example Output:

movie: Fakkah Fuzz: Almost Banned
description: Stand-up comic Fakkah Fuzz mines cross-cultural humor from the experience of growing up as an outspoken Malay man in Singapore.
metric score: 1.0
actual genre: ['Stand-Up Comedy']
predicted genre: ['Stand-Up Comedy']

movie: Mucize
description: Sent away from his family to a remote mountain town, teacher Mahir helps the villagers build a school – and inspires a path to hope for them all.
metric score: 1.0
actual genre: ['Dramas' 'International Movies']
predicted genre: ['Dramas' 'International Movies']
infiniteoverflow commented 3 years ago

@Jayshah6699 I have fixed the branches issue.