[MINOR][ML] Remove unnecessary default value setting for evaluators.
## What changes were proposed in this pull request? Remove unnecessary default value setting for all evaluators, as we have set them in corresponding _HasXXX_ base classes. ## How was this patch tested? Existing tests. Author: Yanbo Liang <ybliang8@gmail.com> Closes #19262 from yanboliang/evaluation.
This commit is contained in:
parent
8319432af6
commit
2f962422a2
|
@ -146,8 +146,7 @@ class BinaryClassificationEvaluator(JavaEvaluator, HasLabelCol, HasRawPrediction
|
||||||
super(BinaryClassificationEvaluator, self).__init__()
|
super(BinaryClassificationEvaluator, self).__init__()
|
||||||
self._java_obj = self._new_java_obj(
|
self._java_obj = self._new_java_obj(
|
||||||
"org.apache.spark.ml.evaluation.BinaryClassificationEvaluator", self.uid)
|
"org.apache.spark.ml.evaluation.BinaryClassificationEvaluator", self.uid)
|
||||||
self._setDefault(rawPredictionCol="rawPrediction", labelCol="label",
|
self._setDefault(metricName="areaUnderROC")
|
||||||
metricName="areaUnderROC")
|
|
||||||
kwargs = self._input_kwargs
|
kwargs = self._input_kwargs
|
||||||
self._set(**kwargs)
|
self._set(**kwargs)
|
||||||
|
|
||||||
|
@ -224,8 +223,7 @@ class RegressionEvaluator(JavaEvaluator, HasLabelCol, HasPredictionCol,
|
||||||
super(RegressionEvaluator, self).__init__()
|
super(RegressionEvaluator, self).__init__()
|
||||||
self._java_obj = self._new_java_obj(
|
self._java_obj = self._new_java_obj(
|
||||||
"org.apache.spark.ml.evaluation.RegressionEvaluator", self.uid)
|
"org.apache.spark.ml.evaluation.RegressionEvaluator", self.uid)
|
||||||
self._setDefault(predictionCol="prediction", labelCol="label",
|
self._setDefault(metricName="rmse")
|
||||||
metricName="rmse")
|
|
||||||
kwargs = self._input_kwargs
|
kwargs = self._input_kwargs
|
||||||
self._set(**kwargs)
|
self._set(**kwargs)
|
||||||
|
|
||||||
|
@ -297,8 +295,7 @@ class MulticlassClassificationEvaluator(JavaEvaluator, HasLabelCol, HasPredictio
|
||||||
super(MulticlassClassificationEvaluator, self).__init__()
|
super(MulticlassClassificationEvaluator, self).__init__()
|
||||||
self._java_obj = self._new_java_obj(
|
self._java_obj = self._new_java_obj(
|
||||||
"org.apache.spark.ml.evaluation.MulticlassClassificationEvaluator", self.uid)
|
"org.apache.spark.ml.evaluation.MulticlassClassificationEvaluator", self.uid)
|
||||||
self._setDefault(predictionCol="prediction", labelCol="label",
|
self._setDefault(metricName="f1")
|
||||||
metricName="f1")
|
|
||||||
kwargs = self._input_kwargs
|
kwargs = self._input_kwargs
|
||||||
self._set(**kwargs)
|
self._set(**kwargs)
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue