|
import streamlit as st |
|
from logic import * |
|
|
|
st.title('ArangoDB - PyG') |
|
|
|
|
|
|
|
st.write('Making Graph') |
|
global movie_mappings, user_mapping, ratings_df, m_id, id_map, sampled_md |
|
movie_mappings, user_mapping, ratings_df, m_id, id_map, sampled_md = make_graph() |
|
st.write('Login to ArangoDB') |
|
login,url,username,password,dbname = login_ArangoDB() |
|
st.write(url) |
|
st.write(username) |
|
st.write(password) |
|
st.write(dbname) |
|
st.write('Loading Graph to ArangoDB') |
|
movie_rec_db = load_data_to_ArangoDB(login) |
|
st.write('Making PyG graph from ArangoDB') |
|
data,train_data, val_data, test_data = make_pyg_graph(movie_rec_db) |
|
st.write('training graph') |
|
model = load_model(train_data, val_data, test_data) |
|
user_id = st.number_input('Insert a number') |
|
ans = get_recommendation(model,data,user_id) |
|
st.write(ans) |
|
|
|
|
