create labels array with unknown
@@ -4,9 +4,9 @@ from utils.Model import Model
def test_augment_data():
- model = Model(10)
+ model = Model(10, labels=['Unknown'])
a = [[1, 2, 3], [4, 5, 6]]
x = np.array(a)
- a = ['label1', 'label2', 'Unknown']
+ a = ['label1', 'label2', 'label3']
y = np.array(a)
model._augment_data(x, y)
