Description
RFormula will index label only when it is string type currently. If the label is numeric type and we use RFormula to present a classification model, there is no label attributes in label column metadata. The label attributes are useful when making prediction for classification, so we can force to index label by StringIndexer whether it is numeric or string type for classification. Then SparkR wrappers can extract label attributes from label column metadata successfully. This feature can help us to fix bug similar with SPARK-15153.
For regression, we will still to keep label as numeric type.
In this PR, we add a param indexLabel to control whether to force to index label for RFormula.
Attachments
Issue Links
- is depended upon by
-
SPARK-15153 SparkR spark.naiveBayes throws error when label is numeric type
-
- Resolved
-
- links to
