navan
/
FlixRec
mirror of https://github.com/navanchauhan/FlixRec.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge branch 'master' of https://github.com/navanchauhan/FlixRec

This commit is contained in:
Navan Chauhan 2022-05-21 10:21:57 -06:00
parent 50be0f1067 5b3ccc0747
commit 078b992f1b
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
db2pc.py Normal file
View File

@ -0,0 +1,16 @@
from database import *
import pandas as pd
from sentence_transformers import SentenceTransformer
database_url = "sqlite:///jlm.db"
engine, Session = init_db_stuff(database_url)
model = SentenceTransformer("paraphrase-multilingual-MiniLM-L12-v2")
df = pd.read_sql("Select * from movies", engine)
df["combined_text"] = df["title"] + ": " + df["overview"].fillna('') + " - " + df["tagline"].fillna('') + " Genres:- " + df["genres"].fillna('')
print(len(df["combined_text"].tolist()))