Giters

dmlc / treelite

Universal model exchange and serialization format for decision tree forests

Home Page:https://treelite.readthedocs.io/en/latest/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Test GTIL with varying input size

hcho3 opened this issue · comments

commented

Use hypothesis to ensure that treelite.gtil.predict produces correct result for inputs of varying sizes. This is important because we use different kernels for small inputs (< 64 rows) and large inputs (> 64 rows)