Esempio n. 1
0
    def setUp(self):
        self.n_train = 200
        self.n_test = 100
        self.contamination = 0.1
        self.roc_floor = 0.8
        self.X_train, self.y_train, self.X_test, self.y_test = generate_data(
            n_train=self.n_train,
            n_test=self.n_test,
            contamination=self.contamination,
            random_state=42)

        detectors = [KNN(), LOF(), OCSVM()]

        self.clf = SimpleDetectorAggregator(base_estimators=detectors,
                                            method='maximization',
                                            contamination=self.contamination)
        self.clf.fit(self.X_train)
Esempio n. 2
0
def construct_simple_aggregator(method):
    from combo.models.detector_comb import SimpleDetectorAggregator
    model = SimpleDetectorAggregator(construct_raw_base_estimators(),
                                     method=method)
    return model