Param for number of epochs (>= 0).
Param for number of epochs (>= 0).
Param for features column name.
Param for features column name.
Param for prediction column name.
Param for prediction column name.
Param for random seed.
Param for random seed.
Param for Column name for predicted class conditional probabilities.
Param for Column name for predicted class conditional probabilities. Note: Not all models output well-calibrated similarity estimates! These probabilities should be treated as confidences, not precise probabilities.
Validates and transforms the input schema.
Validates and transforms the input schema.
input schema
output schema