def setUp(self):
config.StringParameter('test.param0', 'a param', default='foo')
config.StringParameter('test.param1', 'a param', config.oneof('foo', 'bar'))
config.IntParameter('test.param2', 'a param', default=20)
config.IntParameter('test.param3', 'a param', config.atmost(100))
config.IntParameter('test.param4', 'a param', config.atleast(100))
config.IntParameter('test.param5', 'a param', config.between(10,100))
config.IntParameter('test.param6', 'a param', lambda x: x == 50)
config.FloatParameter('test.param7', 'a param', config.between(1.3, 2.7))
_plearnertype = config.StringParameter(
'learner.type',
"""Type of learner to use.
The following learners are included with pebl:
* greedy.GreedyLearner
* simanneal.SimulatedAnnealingLearner
* exhaustive.ListLearner
""",
default = 'greedy.GreedyLearner'
)
_ptasks = config.IntParameter(
'learner.numtasks',
"Number of learner tasks to run.",
config.atleast(0),
default=1
)
class Learner(Task):
def __init__(self, data_=None, prior_=None, whitelist=tuple(), blacklist=tuple(), **kw):
self.data = data_ or data.fromconfig()
self.prior = prior_ or prior.fromconfig()
self.black_edges = kw.pop('blacklist', ())
self.white_edges = kw.pop('whitelist', ())
self.__dict__.update(kw)
# parameters
self.numtasks = config.get('learner.numtasks')
