class A2MLDataset(object):
"""Contains the dataset CRUD operations that interact with provider."""
def __init__(self, ctx, provider):
"""Initializes a new a2ml.
Args:
provider (str): The automl provider/s you wish to run. For example 'auger,azure,google'.
Returns:
A2ML object
"""
super(A2MLDataset, self).__init__()
self.ctx = ctx
self.runner = CRUDRunner(ctx, provider, 'dataset')
@show_result
def list(self):
return self.runner.execute('list')
@show_result
def create(self, source=None):
return self.runner.execute('create', source)
@show_result
def delete(self, name=None):
return self.runner.execute('delete', name)
@show_result
def select(self, name=None):
return self.runner.execute('select', name)
class A2MLExperiment(object):
def __init__(self, ctx, provider):
super(A2MLExperiment, self).__init__()
self.ctx = ctx
self.runner = CRUDRunner(ctx, provider, 'experiment')
@show_result
def list(self):
return self.runner.execute('list')
@show_result
def start(self):
return self.runner.execute('start')
@show_result
def stop(self):
return self.runner.execute('stop')
@show_result
def leaderboard(self, run_id):
return self.runner.execute('leaderboard', run_id)
@show_result
def history(self):
return self.runner.execute('history')
class A2MLModel(metaclass=ErrorHandler):
def __init__(self, ctx, provider):
super(A2MLModel, self).__init__()
self.ctx = ctx
self.runner = CRUDRunner(ctx, provider, 'model')
def deploy(self, model_id, locally):
self.runner.execute('deploy', model_id, locally)
def predict(self, filename, model_id, threshold, locally):
self.runner.execute('predict', filename, model_id, threshold, locally)
class A2MLModel(object):
def __init__(self, ctx, provider):
super(A2MLModel, self).__init__()
self.ctx = ctx
self.runner = CRUDRunner(ctx, provider, 'model')
@show_result
def deploy(self, model_id, locally):
return self.runner.execute('deploy', model_id, locally)
@show_result
def predict(self, filename, model_id, threshold, locally):
return self.runner.execute('predict', filename, model_id, threshold,
locally)
@show_result
def actual(self, filename, model_id):
return self.runner.execute('actual', filename, model_id)
class A2MLExperiment(metaclass=ErrorHandler):
def __init__(self, ctx, provider):
super(A2MLExperiment, self).__init__()
self.ctx = ctx
self.runner = CRUDRunner(ctx, provider, 'experiment')
def list(self):
self.runner.execute('list')
def start(self):
self.runner.execute('start')
def stop(self):
self.runner.execute('stop')
def leaderboard(self, run_id):
self.runner.execute('leaderboard', run_id)
def history(self):
self.runner.execute('history')
class A2MLProject(object):
def __init__(self, ctx, provider):
super(A2MLProject, self).__init__()
self.ctx = ctx
self.runner = CRUDRunner(ctx, provider, 'project')
@show_result
def list(self):
return self.runner.execute('list')
@show_result
def create(self, name):
return self.runner.execute('create', name)
@show_result
def delete(self, name):
return self.runner.execute('delete', name)
@show_result
def select(self, name):
return self.runner.execute('select', name)
class A2MLProject(metaclass=ErrorHandler):
def __init__(self, ctx, provider):
super(A2MLProject, self).__init__()
self.ctx = ctx
self.runner = CRUDRunner(ctx, provider, 'project')
def list(self):
self.runner.execute('list')
def create(self, name):
self.runner.execute('create', name)
def delete(self, name):
self.runner.execute('delete', name)
def select(self, name):
self.runner.execute('select', name)