From ce2536b31fad100ef0ed3e01d85c5ea51e198c96 Mon Sep 17 00:00:00 2001 From: jwradhe Date: Thu, 24 Oct 2024 23:24:14 +0200 Subject: [PATCH] Creating class load_data() --- README.md | 4 ++++ main.py | 8 +++++++- 2 files changed, 11 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index bd85898..a7a47a4 100644 --- a/README.md +++ b/README.md @@ -1,2 +1,6 @@ # Python_AI_Projekt Projekt för Pythonkurs (AI) + + +Autoencoders +Scikit-learn: - K-nearest neighbors \ No newline at end of file diff --git a/main.py b/main.py index 4f71cfd..4470789 100644 --- a/main.py +++ b/main.py @@ -1,6 +1,8 @@ import pandas as pd import re import os +from sklearn.feature_extraction.text import TfidfVectorizer +from sklearn.metrics.pairwise import cosine_similarity class Load_Data: @@ -112,7 +114,11 @@ def main(): data_loader = Load_Data() data = data_loader.check_data() - print(data.head(100)) + if data is not None and not data.empty: + user_input = input("Which Movie or TV-Series do you prefer?: ") + + else: + print("No data available to search.") if __name__ == "__main__": main() \ No newline at end of file