Ну основная причина радости от rtx, насколько я понял это то, что при переходе с 32bit float на 16bit float у людей модельки на 16 гигов в 8гиговы карточки влезают
predict_proba возвращает Two-dimensional numpy.ndarray of shape (number_of_objects, number_of_classes) with the probability for every class for each object, а как понять в какой колнке каой класс (где 0 а где 1 в случае бинарной кл.)?