Python ConeEstimatorGradientの例

コード例 #1

ファイル: ConeEstimator.py プロジェクト: daoudclarke/learncone

 def fit(self, data, class_values):
     if len(set(class_values)) > 2:
         subclassifier = ConeEstimatorGradient()
         subclassifier.set_params(**self.get_params())
         self.classifier = OneVsRestClassifier(subclassifier)
     else:
         self.classifier = ConeEstimatorGradient()
         self.classifier.set_params(**self.get_params())
     self.classifier.fit(data, class_values)

コード例 #2

ファイル: ConeEstimator.py プロジェクト: daoudclarke/learncone

class ConeEstimator(ConeEstimatorBase):
    def fit(self, data, class_values):
        if len(set(class_values)) > 2:
            subclassifier = ConeEstimatorGradient()
            subclassifier.set_params(**self.get_params())
            self.classifier = OneVsRestClassifier(subclassifier)
        else:
            self.classifier = ConeEstimatorGradient()
            self.classifier.set_params(**self.get_params())
        self.classifier.fit(data, class_values)


    def predict(self, data):
        return self.classifier.predict(data)