com.salesforce.op.stages.impl.classification
Function that fits the binary model
Function that fits the binary model
the predictor to wrap
the predictor to wrap
Suggested depth for treeAggregate (greater than or equal to 2).
Suggested depth for treeAggregate (greater than or equal to 2). If the dimensions of features or the number of partitions are large, this param could be adjusted to a larger size. Default is 2.
Set the ElasticNet mixing parameter.
Set the ElasticNet mixing parameter. For alpha = 0, the penalty is an L2 penalty. For alpha = 1, it is an L1 penalty. For alpha in (0,1), the penalty is a combination of L1 and L2. Default is 0.0 which is an L2 penalty.
Note: Fitting under bound constrained optimization only supports L2 regularization, so throws exception if this param is non-zero value.
Sets the value of param family.
Sets the value of param family. Default is "auto".
Whether to fit an intercept term.
Whether to fit an intercept term. Default is true.
Set the lower bounds on coefficients if fitting under bound constrained optimization.
Set the lower bounds on intercepts if fitting under bound constrained optimization.
Set the maximum number of iterations.
Set the maximum number of iterations. Default is 100.
Set the regularization parameter.
Set the regularization parameter. Default is 0.0.
Whether to standardize the training features before fitting the model.
Whether to standardize the training features before fitting the model. The coefficients of models will be always returned on the original scale, so it will be transparent for users. Note that with/without standardization, the models should be always converged to the same solution when no regularization is applied. In R's GLMNET package, the default behavior is true as well. Default is true.
Set the convergence tolerance of iterations.
Set the convergence tolerance of iterations. Smaller value will lead to higher accuracy at the cost of more iterations. Default is 1E-6.
Set the upper bounds on coefficients if fitting under bound constrained optimization.
Set the upper bounds on intercepts if fitting under bound constrained optimization.
Sets the value of param weightCol.
Sets the value of param weightCol. If this is not set or empty, we treat all instance weights as 1.0. Default is not set, so all instances have weight one.
stage uid
stage uid
Wrapper around spark ml logistic regression org.apache.spark.ml.classification.LogisticRegression