Use floats everywhere to get rid of warnings when using sample_weight