thx for the info. I'd be interested in getting the full predict_proba like in scikit learn ( http://scikit-learn.org/stable/modules/generated/sklearn.ensemble.RandomForestClassifier.html#sklearn.ensemble.RandomForestClassifier.predict_proba) for the random forest model. There doesn't seem to be a way to get the details, is there any reason for that ?
Regards, Olivier. Le lun. 13 juil. 2015 à 21:12, Feynman Liang <fli...@databricks.com> a écrit : > There is MulticlassMetrics in MLlib; unfortunately a pipelined version > hasn't yet been made for spark-ml. SPARK-7690 > <https://issues.apache.org/jira/browse/SPARK-7690> is tracking work on > this if you are interested in following the development. > > On Mon, Jul 13, 2015 at 2:16 AM, Olivier Girardot < > o.girar...@lateral-thoughts.com> wrote: > >> Hi everyone, >> Using spark-ml there seems to be only BinaryClassificationEvaluator and >> RegressionEvaluator, is there any way or plan to provide a ROC-based or >> PR-based or F-Measure based for multi-class, I would be interested >> especially in evaluating and doing a grid search for a RandomForest model. >> >> Regards, >> >> Olivier. >> > >
