| import streamlit as st | |
| from logic import * | |
| st.title('ArangoDB - PyG') | |
| #--------------------- | |
| st.write('Making Graph') | |
| movie_mappings, user_mapping, ratings_df, m_id, id_map, sampled_md = make_graph() | |
| st.write('Login to ArangoDB') | |
| login = login_ArangoDB() | |
| st.write('Loading Graph to ArangoDB') | |
| movie_rec_db = load_data_to_ArangoDB(login) | |
| st.write('Making PyG graph from ArangoDB') | |
| train_data, val_data, test_data = make_pyg_graph(movie_rec_db) | |
| st.write('training graph') | |
| train(train_data, val_data, test_data) | |
| #--------------------- | |