示例#1
0
    def train(self, arr, remember=True,
              regression_type="Ridge"):
        f = Features()

        if os.path.exists(self.filename):
            self.clf = self._load_clf()
            train_X, train_y = f.proportion_class(arr)
        else:
            self.clf = self.new_clf(regression_type=regression_type)
            train_X, train_y = f.proportion_class(arr, len(arr))

        self.clf.fit(train_X, train_y)
        if remember:
            self._save_clf()

        return train_X, train_y