File size: 222 Bytes
bbd259b |
1 2 3 4 5 6 7 8 9 10 11 |
import numpy as np
from src.train_classifier import train_and_save
# DUMMY FEATURES (9 features as defined)
X = np.random.rand(20, 9)
# DUMMY LABELS (5 classes)
y = np.random.randint(0, 5, size=20)
train_and_save(X, y)
|