PPPLDeepLearning / plasma-python

PPPL deep learning disruption prediction package
http://tigress-web.princeton.edu/~alexeys/docs-web/html/
79 stars 43 forks source link

Yaml-based config #3

Closed ASvyatkovskiy closed 7 years ago

ASvyatkovskiy commented 7 years ago

The pull request contains following changes:

    with open(input_file, 'r') as yaml_file:
        params = yaml.load(yaml_file)
        ....

For instance, the conf['data']['target'] parameter is not an object, but a string. A proper object is created based on the string matching (similarly to the Normalizers):

        if params['target'] == 'hinge':
            params['data']['target'] = t.HingeTarget
        elif params['target'] == 'binary':
            params['data']['target'] = t.BinaryTarget
        elif params['target'] == 'ttd':
            params['data']['target'] = t.TTDTarget
        elif params['target'] == 'ttdlinear':
            params['data']['target'] = t.TTDLinearTarget
        else:
            print('Unkown type of target. Exiting')
            exit(1)