def predict_proba(self, X): if isinstance(X, pd.DataFrame): X = X.values instances, _ = sklearn_input_to_weka( X, labels=self._labels) distributions = self._weka_model.distributions_for_instances(instances) return distributions
def fit(self, X, y): options = self.weka_options() self._weka_model = weka.classifiers.Classifier( classname='weka.classifiers.trees.J48', options=options) if isinstance(X, pd.DataFrame): X = X.values instances, labels = sklearn_input_to_weka(X, y) self._weka_model.build_classifier(instances) self._labels = labels return self