Beispiel #1
0
def main(filename):
    logger.info("================ Starting machine learning model process ================")

    try:
        utils = Utils()
        models = Model()

        df = utils._read_from_csv(filename)
        transform = Transform(df,5,2)
        features,target = transform._transform_process()
        x_train,x_test,y_train,y_test = utils._split_train_test(features,target)

        model = models._training_best_model(x_train,y_train,x_test,y_test)
        models._make_predictions(model,x_test,y_test)
    except FileNotFoundError:
        logger.error('File not found for {}'.format(filename))