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)