steveschulze / ml_workshop

ml_workshop

We now use this for the first stage of classification for the ELAsTiCC challenge.

Useful links:

• ELAsTiCC documentation
• Data

How To:

import os, logging
from train_models import Model

logging.basicConfig()
logger = logging.getLogger()
logger.setLevel(logging.INFO)

path_to_trainingset = os.path.join(
    os.path.dirname(os.path.realpath(__file__)),
    "data",
    "elasticc_feature_trainingset_v3",
)

m = Model(
    stage="1",
    path_to_trainingset=path_to_trainingset,
    n_iter=10,
    random_state=42,
    one_alert_per_stock=True,
)
m.split_sample()
m.train()
m.evaluate()