[
https://issues.apache.org/jira/browse/SPARK-20043?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=15939571#comment-15939571
]
Yan Facai (颜发才) commented on SPARK-20043:
-----------------------------------------
The bug can be reproduced by:
```scala
test("cross validation with decision tree") {
val dt = new DecisionTreeClassifier()
val dtParamMaps = new ParamGridBuilder()
.addGrid(dt.impurity, Array("Gini", "Entropy"))
.build()
val eval = new BinaryClassificationEvaluator
val cv = new CrossValidator()
.setEstimator(dt)
.setEstimatorParamMaps(dtParamMaps)
.setEvaluator(eval)
.setNumFolds(3)
val cvModel = cv.fit(dataset)
// copied model must have the same paren.
val cv2 = testDefaultReadWrite(cvModel, testParams = false)
}
```
I'd like to work on it.
> CrossValidatorModel loader does not recognize impurity "Gini" and "Entropy"
> on ML random forest and decision. Only "gini" and "entropy" (in lower case)
> are accepted
> --------------------------------------------------------------------------------------------------------------------------------------------------------------------
>
> Key: SPARK-20043
> URL: https://issues.apache.org/jira/browse/SPARK-20043
> Project: Spark
> Issue Type: Bug
> Components: ML
> Affects Versions: 2.1.0
> Reporter: Zied Sellami
> Labels: starter
>
> I saved a CrossValidatorModel with a decision tree and a random forest. I use
> Paramgrid to test "gini" and "entropy" impurity. CrossValidatorModel are not
> able to load the saved model, when impurity are written not in lowercase. I
> obtain an error from Spark "impurity Gini (Entropy) not recognized.
--
This message was sent by Atlassian JIRA
(v6.3.15#6346)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
