def test_build_learner_config(self):
     experimenter = Experimenter(learner_config={'vw': {'param': ['1', '2']}})
     experimenter.build_learner_config()
     self.assertEquals(experimenter.learner_config, {'vw': [{'param': '1'}, {'param': '2'}]})
     experimenter = Experimenter(
         learner_config={'vw': {'param1': ['1', '2'], 'param2': [3, 4]}})
     experimenter.build_learner_config()
     print 'learner_config', experimenter.learner_config
     self.assertEquals(experimenter.learner_config,
                       {'vw': [{'param1': '1', 'param2': 3}, {'param1': '2', 'param2': 3},
                               {'param1': '1', 'param2': 4}, {'param1': '2', 'param2': 4}]})
     experimenter = Experimenter(
         learner_config={'vw': {'param1': ['1', '2'], 'param2': [3, 4]}, 'fm': {'param3': ['5']}})
     experimenter.build_learner_config()
     print 'learner_config', experimenter.learner_config
     self.assertEquals(experimenter.learner_config,
                       {'vw': [{'param1': '1', 'param2': 3}, {'param1': '2', 'param2': 3},
                               {'param1': '1', 'param2': 4}, {'param1': '2', 'param2': 4}],
                        'fm': [{'param3': '5'}]})
Ejemplo n.º 2
0
 def test_build_learner_config(self):
     experimenter = Experimenter(
         learner_config={'vw': {
             'param': ['1', '2']
         }})
     experimenter.build_learner_config()
     self.assertEquals(experimenter.learner_config,
                       {'vw': [{
                           'param': '1'
                       }, {
                           'param': '2'
                       }]})
     experimenter = Experimenter(
         learner_config={'vw': {
             'param1': ['1', '2'],
             'param2': [3, 4]
         }})
     experimenter.build_learner_config()
     print 'learner_config', experimenter.learner_config
     self.assertEquals(
         experimenter.learner_config, {
             'vw': [{
                 'param1': '1',
                 'param2': 3
             }, {
                 'param1': '2',
                 'param2': 3
             }, {
                 'param1': '1',
                 'param2': 4
             }, {
                 'param1': '2',
                 'param2': 4
             }]
         })
     experimenter = Experimenter(
         learner_config={
             'vw': {
                 'param1': ['1', '2'],
                 'param2': [3, 4]
             },
             'fm': {
                 'param3': ['5']
             }
         })
     experimenter.build_learner_config()
     print 'learner_config', experimenter.learner_config
     self.assertEquals(
         experimenter.learner_config, {
             'vw': [{
                 'param1': '1',
                 'param2': 3
             }, {
                 'param1': '2',
                 'param2': 3
             }, {
                 'param1': '1',
                 'param2': 4
             }, {
                 'param1': '2',
                 'param2': 4
             }],
             'fm': [{
                 'param3': '5'
             }]
         })