faction-picker-bot/.ipynb_checkpoints/train_models-checkpoint.py

import pandas as pd
import numpy as np
import lightgbm as lgb
import argparse
import pickle
import yaml
import math
import time


def main(pickledir):
    with open(pickledir, 'rb') as fd:
        each_faction_dataset = pickle.load(fd)

    for faction in each_faction_dataset.keys():
        X_data = lgb.Dataset(each_faction_dataset[faction]['features'])
        y_data = each_faction_dataset[faction]['vp']



if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='Input DVC params.')
    parser.add_argument('--params', type=str)
    args = parser.parse_args()
    paramsdir = args.params

    with open(paramsdir, 'r') as fd:
        params = yaml.safe_load(fd)

    pickledir = params['prepare-step2']['pickle-dir']

    main(pickledir)