tomasonjo / blogs

Jupyter notebooks that support my graph data science blog posts at https://bratanic-tomaz.medium.com/
1.29k stars 334 forks source link

Update Movie_recommendations.ipynb #8

Closed mfyuce closed 1 year ago

mfyuce commented 1 year ago

fix for: TypeError: HeteroData.get_edge_store() missing 1 required positional argument: 'dst' in weight = torch.bincount(train_data['user', 'movie'].edge_label)


TypeError Traceback (most recent call last) Cell In[48], line 1 ----> 1 weight = torch.bincount(train_data['user', 'movie'].edge_label) 2 weight = weight.max() / weight 4 def weighted_mse_loss(pred, target, weight=None):

File ~/TorchStudio/python/envs/pyg/lib/python3.10/site-packages/torch_geometric/data/hetero_data.py:164, in HeteroData.getitem(self, args) 161 return out 163 if isinstance(key, tuple): --> 164 return self.get_edge_store(key) 165 else: 166 return self.get_node_store(key)

TypeError: HeteroData.get_edge_store() missing 1 required positional argument: 'dst'

tomasonjo commented 1 year ago

Thanks