def test_maxklpe():
"""Check MaxKLPE"""
score_train_true = -np.array([np.sqrt(10), 2, np.sqrt(10)])
pred_train_true = np.array([0, 1, 0])
score_test_true = -np.array([np.sqrt(5), 2])
pred_test_true = np.array([1, 1])
# when novelty=False, i.e. scores and predict on X_train itself
clf1 = MaxKLPE(k=2, contamination=0.7)
clf1.fit(X_train)
assert_equal(clf1.algo, 'max')
score_train_attr1 = clf1.scores_fit_
assert_array_almost_equal(score_train_attr1, score_train_true)
score_train1 = clf1.score_samples(X_train)
assert_array_almost_equal(score_train1, score_train_true)
assert_array_equal((score_train1 >= clf1.threshold_).astype(int),
clf1.predict(X_train))
assert_array_equal(pred_train_true, clf1.predict(X_train))
# when novelty=True, i.e. scores and predict on X_test
clf2 = MaxKLPE(k=2, contamination=0.7, novelty=True)
clf2.fit(X_train)
score_train_attr2 = clf2.scores_fit_
assert_array_almost_equal(score_train_attr2, score_train_true)
score_test2 = clf2.score_samples(X_test)
assert_array_almost_equal(score_test2, score_test_true)
assert_array_equal(pred_test_true, clf2.predict(X_test))
def test_klpe_contamination():
"""Check that predict agrees with contamination parameter. """
# This requires a certain amount of data samples because the threshold is
# defined by a quantile.
X = np.random.randn(50, 2)
contamination = 0.1
clf1 = AverageKLPE(k=5, contamination=contamination)
clf1.fit(X)
assert_almost_equal(np.mean(clf1.predict(X) == 1), 1 - contamination)
clf2 = MaxKLPE(k=5, contamination=contamination)
clf2.fit(X)
assert_almost_equal(np.mean(clf2.predict(X) == 1), 1 - contamination)
def test_score_train_novelty_or_not():
"""Check score_fit_ attribute is the same if novelty=True of False"""
X = np.random.randn(50, 2)
# for AverageKLPE
clf1 = AverageKLPE(k=10)
clf2 = AverageKLPE(k=10, novelty=True)
clf1.fit(X)
clf2.fit(X)
assert_array_equal(clf1.scores_fit_, clf2.scores_fit_)
# for MaxKLPE
clf3 = MaxKLPE(k=10)
clf4 = MaxKLPE(k=10, novelty=True)
clf3.fit(X)
clf4.fit(X)
assert_array_equal(clf3.scores_fit_, clf4.scores_fit_)