Update logic.py
Browse files
logic.py
CHANGED
|
@@ -348,6 +348,7 @@ def make_pyg_graph(movie_rec_db):
|
|
| 348 |
rev_edge_types=[('movie', 'rev_rates', 'user')],
|
| 349 |
)(data)
|
| 350 |
|
|
|
|
| 351 |
|
| 352 |
|
| 353 |
|
|
|
|
| 348 |
rev_edge_types=[('movie', 'rev_rates', 'user')],
|
| 349 |
)(data)
|
| 350 |
|
| 351 |
+
return train_data, val_data, test_data
|
| 352 |
|
| 353 |
|
| 354 |
|