Example #1
0
 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
Example #2
0
 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