• Quan (Andy) Gan's avatar
    [Feature] Edge DataLoader for edge classification & link prediction (#1828) · f5eb80d2
    Quan (Andy) Gan authored
    * clean commit
    
    * oops forgot the most important files
    
    * use einsum
    
    * copy feature from frontier to block
    
    * Revert "copy feature from frontier to block"
    
    This reverts commit 5224ec963eb6a3ef1b6ab74d8ecbd44e4e42f285.
    
    * temp fix
    
    * unit test
    
    * fix
    
    * revert jtnn
    
    * lint
    
    * fix win64
    
    * docstring fixes and doc indexing
    
    * revert einsum in sparse bidecoder
    
    * fix some examples
    
    * lint
    
    * fix due to some tediousness in remove_edges
    
    * addresses comments
    
    * fix
    
    * more jtnn fixes
    
    * fix
    f5eb80d2
test_transform.py 60.3 KB