sklearn_wrappers.py 文件源码

def __init__(self, data, N_i, N_c, *args, **kwargs):
        """ Fit a random forest model to a Dataset object.

        N_i, N_c: parameters defining allowed time windows. See the
        transform_X method.

        args, kwargs: passed to the LogisticRegressionCV constructor.

        """
        Wrapper.__init__(self,data,N_i,N_c)
        kwargs['n_estimators'] = 128
        self.classifier = RandomForestClassifier(*args, **kwargs)
        self.classifier.fit(self.fit_X,self.fit_y)