search

shiba24 / learning2rank

Learning to rank with neuralnet - RankNet and ListNet
481 stars 141 forks source link

type-cast N_train to integer #9

Closed akashrajkn closed 6 years ago

akashrajkn commented 6 years ago

numpy.split requires indices_or_sections to be of type int

shiba24 commented 6 years ago

Thank you @akashrajkn for the commit! Could you tell me the numpy version? I think my code (on current master) might depend on older numpy, so just would like to clarify that.

akashrajkn commented 6 years ago

@shiba24 NumPy v1.13 and above