pytorch_cluster
pytorch_cluster copied to clipboard
rusty1s
PyTorch Extension Library of Optimized Graph Cluster Algorithms
In my work I find myself making frequent cuda calls to `torch_cluster.nearest` with the form `nearest(different_every_time, same_every_time)` without providing a `batch_x` or `batch_y`. `different_every_time` is on the order of say...
This commit implements weighted biased random walks as in the original Node2vec paper. In particular, it adds a new parameter to the `random_walk` function, i.e., `edge_weight`, which allows passing edge...
In the Node2vec paper, the transition probabilities are computed based on both edge weights and the hyperparameters `p` and `q`. The current implementation does not include the edge weights of...
Hi, I'm trying to get the edge attributes for each step in a random walk, with the use case of trying to retain edge types in a walk on a...
Hi there, is there any chance that you will add torch as a requirement to setup.py or adopt [PEP 517](https://peps.python.org/pep-0517/)? I am running into problems installing this package while trying...
Related: https://github.com/pytorch/pytorch/pull/31378 Current implementation: https://github.com/rusty1s/pytorch_cluster/blob/dbcafbe6a60aaa631b39050e3aa228f6d3fd1592/csrc/cuda/knn_cuda.cu#L56 As introduced in the linked PR for PyTorch, normalizing the vectors first and then doing the inner product could prevent overflow for vectors with large...
**TL;DR** Add a switch for `knn` and `knn_graph` to enable returning distances alongside the edge_index. This code has been tested on CPU and CUDA 11.3 with PyTorch 1.11.0 on Ubuntu...
I'd like to load a TorchScript model trained with Python and which uses 4 libs: `pytorch-geometric`, `pytorch-scatter`, `pytorch-sparse` and `pytorch-cluster`. The export part is ok. Here is my minimal python...
Hi, I tried to install your project with conda running `conda install -c pyg pytorch-cluster` but it did not work. Here is the overall output message: ``` Collecting package metadata...
The current _random walk_ function lacks one specific parameter(restart ratio). Will this parameter be added later?
