jwradhe/TV-Show-recommender
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Uppdatera README samt kommentar i trainmodel

Browse Source
This commit is contained in:
jwradhe 2024-11-06 22:58:36 +01:00
parent 5c9f0ffdc8
commit a26b150a19
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
README.md
View File

@ -27,7 +27,8 @@ I will use a dataset from TMBD
https://www.kaggle.com/datasets/asaniczka/full-tmdb-tv-shows-dataset-2023-150k-shows
### Model:
I will use NearestNeighbors (NN) alhorithm together with K-NearestNeighbors alhorithm.
I must first preprocess data with vectorization so that i can use it in NearestNeighbors (NN) alhorithm.
I will use NearestNeighbors (NN) output as input to K-NearestNeighbors alhorithm.
### Features:
1. Load data from dataset and preprocessing.

4
trainmodel.py
View File

@ -34,7 +34,7 @@ class TrainModel:
# Preprocess title data
preproccessed_data = self.preprocess_title_data()
# Train the NN model
# Train the NearestNeighbors model
self.model.fit(preproccessed_data)
stop = time.time()
@ -51,7 +51,7 @@ class TrainModel:
# Preprocess target data
target_vector = self.preprocess_target_data(target_row)
# Get nearest neighbors
# Use NearestNeighbors model as input to K-nearest neighbors
distances, indices = self.model.kneighbors(target_vector, n_neighbors=num_recommendations)
recommendations = self.title_data.iloc[indices[0]].copy()
recommendations['distance'] = distances[0]