You are viewing a plain text version of this content. The canonical link for it is here.

Posted to issues@spark.apache.org by "holdenk (JIRA)" <ji...@apache.org> on 2015/07/02 02:20:04 UTC

[jira] [Commented] (SPARK-8069) Add support for cutoff to RandomForestClassifier

    [ https://issues.apache.org/jira/browse/SPARK-8069?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=14611224#comment-14611224 ] 

holdenk commented on SPARK-8069:
--------------------------------

So I started working on doing this for the randomforestclassifier. I think we could maybe add this as a trait for multiclass classification models which return all of the scores (but the current API of the multiclass classification models seems to just return the winning class). What about about if we added a trait to do this and then the other models could implement it as needed? Or I could change the base predictionmodel but that would probably be a pretty large change.

> Add support for cutoff to RandomForestClassifier
> ------------------------------------------------
>
>                 Key: SPARK-8069
>                 URL: https://issues.apache.org/jira/browse/SPARK-8069
>             Project: Spark
>          Issue Type: Improvement
>          Components: ML
>            Reporter: holdenk
>            Priority: Minor
>   Original Estimate: 240h
>  Remaining Estimate: 240h
>
> Consider adding support for cutoffs similar to http://cran.r-project.org/web/packages/randomForest/randomForest.pdf 



--
This message was sent by Atlassian JIRA
(v6.3.4#6332)

---------------------------------------------------------------------
To unsubscribe, e-mail: issues-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org