test_iforest.py 文件源码

def test_recalculate_max_depth():
    """Check that max_depth is recalculated when max_samples is reset to n_samples"""
    X = iris.data
    clf = IsolationForest().fit(X)
    for est in clf.estimators_:
        assert_equal(est.max_depth, int(np.ceil(np.log2(X.shape[0]))))