In [2]:

    

import yaml


    

In [9]:

    

import yaml
def read_model_yaml(inp_yaml):    
    # with open("../model/spec/SS/SS_RF_1.yaml") as stream:    
    with open(inp_yaml) as stream:    
        data = yaml.load(stream)
        parameters_dict = {
        "author"                   : data["author"]
        , "train_data_source"      : data["train_data_source"]
        , "test_data_source"       : data["test_data_source"]
        , "holdout_data_source"    : data["holdout_data_source"]
        , "label_colname"          : data["label_colname"]
        , "simulations"            : data["simulations"]
        , "predict"                : data["predict"]
        , "model_type"             : data["model_type"]
        , "parameters"             : data[data["model_type"]]
        }
    return parameters_dict


    

In [10]:

    

a = read_model_yaml("../model/spec/SS/SS_RF_1.yaml")


    

In [11]:

    

a.keys()


    

    
Out[11]:





dict_keys(['predict', 'train_data_source', 'author', 'simulations', 'test_data_source', 'model_type', 'holdout_data_source', 'parameters', 'label_colname'])

In [17]:

    

a["parameters"]["criterion"]


    

    
Out[17]:





{'criterion': 'gini',
 'max_depth': 2,
 'max_features': 'auto',
 'n_estimators': 10,
 'n_jobs': 1}

In [19]:

    

type(a["simulations"])


    

    
Out[19]:





bool

In [ ]: