• Jeremy Reizenstein's avatar
    heterogenous KNN · 01b5f7b2
    Jeremy Reizenstein authored
    Summary: Interface and working implementation of ragged KNN. Benchmarks (which aren't ragged) haven't slowed. New benchmark shows that ragged is faster than non-ragged of the same shape.
    
    Reviewed By: jcjohnson
    
    Differential Revision: D20696507
    
    fbshipit-source-id: 21b80f71343a3475c8d3ee0ce2680f92f0fae4de
    01b5f7b2
test_knn.py 4.64 KB