Hugging Face's logo

FabiInSpringday
/
Netflix-Movie-TVShow-Classifier

Text Classification
Transformers
English
distilbert
netflix
movie
tv-show
Model card Files
xet
 Community
Netflix-Movie-TVShow-Classifier
File size: 1,725 Bytes
f865031
 
 
 
 
 
 
 
 
 
 
 
 
 
1d7513d
 
 
 
b9413ff
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
---
language:
- en
license: apache-2.0
library_name: transformers
pipeline_tag: text-classification
base_model: distilbert/distilbert-base-uncased
tags:
- distilbert
- netflix
- text-classification
- movie
- tv-show
---




# Netflix Movie-TV Show Classifier

## Model Description

This project uses a fine-tuned DistilBERT model to classify Netflix content as either a Movie or a TV Show.

## Base Model

distilbert-base-uncased

## Dataset

Netflix Titles Dataset (`netflix_titles.csv`)

The dataset contains information about Netflix movies and TV shows, including:

- Title
- Description
- Genre
- Director
- Cast
- Country
- Release Year
- Rating
- Type

## Model Input

The model uses:

Title + Description + Genre

Example:

"The Lost Kingdom. A young warrior travels across an ancient kingdom to rescue his family. Action, Adventure."

## Output Classes

- Movie
- TV Show

## Fine-Tuning

The pre-trained DistilBERT model was fine-tuned using the Netflix dataset for binary text classification.

Training settings:

- Epochs: 3
- Learning Rate: 2e-5
- Batch Size: 16
- Maximum Sequence Length: 256
- Optimizer: AdamW

## Prediction Example

Input:

Title: The Lost Kingdom

Description: A young warrior travels across an ancient kingdom to rescue his family and defeat a powerful enemy.

Genre: Action, Adventure

Output:

Prediction: Movie

## Evaluation

The model is evaluated using:

- Accuracy
- Precision
- Recall
- F1-score

## Technology Used

- Python
- PyTorch
- Hugging Face Transformers
- DistilBERT
- Pandas
- Scikit-learn
- Google Colab

## Purpose

The purpose of this project is to demonstrate how a pre-trained transformer model can be fine-tuned on a custom Netflix dataset for text classification.