Muhammed Fatih BALIN
bf264d00fe
* adding LABOR sampling * add ladies and pladies samplers * fix compile error after rebase * add reference for ladies sampler * Improve ladies implementation. * weighted labor sampling initial implementation draft fix indentation and small bug in ladies script * importance_sampling currently doesn't work with weights * fix weighted importance sampling * move labor example into its own folder * lint fixes * Improve documentation * remove examples from the main PR * fix linting by not using c++17 features * fix documentation of labor_sampler.py * update documentation for labor.py * reformat the labor.py file with black * fix linting errors * replace exception use with if * fix typo in error comment * fixing win64 build for ci * fixing weighted implementation, works now. * fix bug in the weighted case and importance_sampling==0 * address part of the reviews * remove unused code paths from cuda * remove unused code path from cpu side * remove extra features of labor making use of random seed. * fix exclude_edges bug * remove pcg and seed logic from cpu implementation, seed logic should still work for cuda. * minor style change * refactor CPU implementation, take out the importance_sampling probability computation into a function. * improve CUDAWorkspaceAllocator * refactor importance_sampling part out to a function * minor optimization * fix linting issue * Revert "remove pcg and seed logic from cpu implementation, seed logic should still work for cuda." This reverts commit c250e07ac6d7e13f57e79e8a2c2f098d777378c2. * Revert "remove extra features of labor making use of random seed." This reverts commit 7f99034353080308f4783f27d9a08bea343fb796. * fix the documentation * disable NIDs * improve the documentation in the code * use the stream argument in pcg32 instead of skipping ahead t times, can discard the use of hashmap now since it is faster this way. * fix linting issue * address another round of reviews * further optimize CPU LABOR sampling implementation * fix linting error * update the comment * reformat * rename and rephrase comment * fix formatting according to new linting specs * fix compile error due to renaming, fix linting. * lint * rename DGLHeteroGraph to DGLGraph to match master * replace other occurrences of DGLHeteroGraph to DGLGraph Co-authored-by: Muhammed Fatih BALIN <m.f.balin@gmail.com> Co-authored-by: Kaan Sancak <kaansnck@gmail.com> Co-authored-by: Quan Gan <coin2028@hotmail.com>
15 lines
517 B
Python
15 lines
517 B
Python
"""The ``dgl.sampling`` package contains operators and utilities for
|
|
sampling from a graph via random walks, neighbor sampling, etc. They
|
|
are typically used together with the ``DataLoader`` s in the
|
|
``dgl.dataloading`` package. The user guide :ref:`guide-minibatch`
|
|
gives a holistic explanation on how different components work together.
|
|
"""
|
|
|
|
from .randomwalks import *
|
|
from .pinsage import *
|
|
from .neighbor import *
|
|
from .labor import *
|
|
from .node2vec_randomwalk import *
|
|
from .negative import *
|
|
from . import utils
|