async def route_label_item(request, dataset_name):
result = resp('success')
dataset = datasets.get(name=dataset_name)
labels = classify(dataset, datasets_bundle, request)
if len(labels) > 0:
result['data'] = labels
else:
result = resp('error')
result['reason'] = "Maybe you have to train this dataset first."
return json(result, status=201)
async def route_train_dataset(request, dataset_name):
result = resp('error')
result['data'] = {}
request_json = request.json
dataset = datasets.get(name=dataset_name)
training_steps = request_json.get('training_steps', 50)
train_task = {
"action": "train",
"dataset": dataset,
"training_steps": training_steps
}
if dataset['trainable'] == True:
TrainWorker(dataset['path'], training_steps)
result = resp('success')
result['data'] = {}
result['data']['task'] = train_task
else:
result['reason'] = "This dataset is not trainable"
return json(result, status=200)
async def route_get_one_dataset(request, dataset_name):
return json(datasets.get(name=dataset_name), status=200)
async def route_get_datasets(request):
return json(datasets.get(), status=200)
