def test_kNN_accuracy_0(self):
"""
The accuracy for this single-point dataset should be 0.
"""
X_train = np.array([[[1]], [[0]]])
y_train = np.array([[1, 0], [0, 1]])
X_val = np.array([[[0.9]]])
y_val = np.array([[0, 1]])
acc = find_architecture.kNN_accuracy(
X_train, y_train, X_val, y_val, k=1)
assert acc == approx(0)
def test_kNN_accuracy_0(self):
"""
The accuracy for this single-point dataset should be 0.
"""
X_train = np.array([[[1]], [[0]]])
y_train = np.array([[1, 0], [0, 1]])
X_val = np.array([[[0.9]]])
y_val = np.array([[0, 1]])
acc = find_architecture.kNN_accuracy(
X_train, y_train, X_val, y_val, k=1)
assert acc == approx(0)
def test_kNN_accuracy_1(self):
"""
The accuracy for this single-point dataset should be 1.
"""
X_train = np.array([[[1]], [[0]]])
y_train = np.array([[1, 0], [0, 1]])
X_val = np.array([[[0.9]]])
y_val = np.array([[1, 0]])
acc = find_architecture.kNN_accuracy(X_train,
y_train,
X_val,
y_val,
k=1)
assert_almost_equal(acc, 1.0)
