logger_test.py 文件源码

def test_file_logger(self):
        X, y = datasets.make_classification(random_state=42)
        task = Task("class_split", X, y, "classification", test_size=0.1, random_state=42)
        scorer = Scorer("auc_error", lambda y_pred, y_true: 1 - metrics.roc_auc_score(y_pred, y_true))
        logger = FileLogger(task)
        optimizer = XGBoostOptimizer(task, scorer, logger)
        optimizer.start_optimization(max_evals=10)
        self.assertEqual(len(list(logger.load_all_results())), 10)
        os.remove(task.name + ".log")