Sampling Issue

[BobbyCaliber]

Пытаюсь просэмплировать данные, но иногда выходит такая ошибка, либо строк данных оказывается меньше, чем указано в аргументе функции. Приложил вид датасета и инициализацию моделей.

[Anaxagor]

Добрый день! Строк меньше бывает, потому что по умолчанию функция удаляет строки с отрицательными значениями в переменных, в которых этого не должно быть. Чтобы это отключить, нужно у функции sample установить флаг filter_neg = False.
По поводу случая с nan посмотрим, есть идеи, почему это происходит, возможно получится в ближайшее время пофиксить.