def test_precision_k_curve_multiple_list_2(self):
"""Test precision_k_curve on a list."""
y_true = [[0, 0, 0, 1, 1], [1, 0, 0, 0, 1]]
y_pred = [[0.1, 0.2, 0.0, 0.0, 0.3], [0.1, 0.0, 0.0, 0.2, 0.3]]
actual = precision_k_curve(y_true, y_pred, max_k=3)
expected = [2 / 2, 2 / 4, 3 / 6]
np.testing.assert_allclose(expected, actual)
def test_precision_k_curve_pd(self):
"""Test precision_k_curve on a pandas data fame."""
y_true = pd.DataFrame([[0, 0, 0, 1, 1]])
y_pred = pd.DataFrame([[0.1, 0.0, 0.0, 0.2, 0.3]])
actual = precision_k_curve(y_true, y_pred, max_k=3)
expected = [1, 1., 2 / 3]
np.testing.assert_allclose(expected, actual)
def test_precision_k_curve_np(self):
"""Test precision_k_curve on a numpy array."""
y_true = np.array([[0, 0, 0, 1, 1]])
y_pred = np.array([[0.1, 0.0, 0.0, 0.2, 0.3]])
actual = precision_k_curve(y_true, y_pred, max_k=3)
expected = [1, 1., 2 / 3]
np.testing.assert_allclose(expected, actual)
def test_precision_k_curve_list_4(self):
"""Test precision_k_curve on a list."""
y_true = [[1, 0, 0, 0, 0, 0, 0, 0, 0, 0]]
y_pred = [[0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0]]
actual = precision_k_curve(y_true, y_pred, max_k=10)
expected = [0, 0, 0, 0, 0, 0, 0, 0, 0, 1 / 10]
np.testing.assert_allclose(expected, actual)
def test_precision_k_curve_list(self):
"""Test precision_k_curve on a list."""
y_true = [[0, 0, 0, 1, 1]]
y_pred = [[0.1, 0.0, 0.0, 0.2, 0.3]]
actual = precision_k_curve(y_true, y_pred, max_k=3)
expected = [1, 1., 2 / 3]
np.testing.assert_allclose(expected, actual)
def test_precision_k_curve_list_3(self):
"""Test precision_k_curve on a list."""
y_true = [[1, 0, 0, 0, 0, 0, 0, 0, 0, 0]]
y_pred = [[0.4, 0.0, 0.0, 0.2, 0.3, 0.0, 0.0, 0.0, 0.0, 0.0]]
actual = precision_k_curve(y_true, y_pred, max_k=10)
expected = [
1, 1 / 2, 1 / 3, 1 / 4, 1 / 5, 1 / 6, 1 / 7, 1 / 8, 1 / 9, 1 / 10
]
np.testing.assert_allclose(expected, actual)
