| 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) | |
| 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) | |