async def delete_tensorboard(container_info: ContainerInfo):
try:
tensorboard_service.tensorboard_delete(container_info)
return ApiResponse(success=True)
except ApplicationError as e:
raise HTTPException(status_code=400, detail=e.__str__())
except Exception as e:
raise HTTPException(status_code=500, detail=e.__str__())
async def create_tfrecord(labels_info: LabelsInformation):
try:
tfrecord_manager.prepare_dataset(labels_info=labels_info)
return ApiResponse(success=True)
except ApplicationError as e:
raise HTTPException(status_code=400, detail=e.__str__())
except Exception as e:
raise HTTPException(status_code=500, detail=e.__str__())
async def validate_dataset(dataset_info: DatasetInfo):
try:
dataset_validator_service.validate_dataset(dataset_info)
return ApiResponse(success=True)
except ApplicationError as e:
raise HTTPException(status_code=400, detail=e.__str__())
except Exception as e:
raise HTTPException(status_code=500, detail=e.__str__())
async def stop_container(container_info: ContainerInfo):
try:
job_management_service.stop_container(container_info=container_info)
return ApiResponse(success=True, data="Job Killed")
except ApplicationError as e:
raise HTTPException(status_code=400, detail=e.__str__())
except Exception as e:
raise HTTPException(status_code=500, detail=e.__str__())
async def start_container(container_settings: ContainerSettings):
try:
job_management_service.start_container(
container_settings=container_settings)
return ApiResponse(success=True, data="Container Started")
except ApplicationError as e:
raise HTTPException(status_code=404, detail=e.__str__())
except Exception as e:
raise HTTPException(status_code=404, detail=e.__str__())
async def refresh_tensorboard():
try:
tensorboard_service.tensorboard_stop()
tensorboard_service.tensorboard_start()
return ApiResponse(success=True)
except ApplicationError as e:
raise HTTPException(status_code=400, detail=e.__str__())
except Exception as e:
raise HTTPException(status_code=500, detail=e.__str__())
async def create_advance_configuration(
configuration_content: ConfigurationContent):
try:
configuration_writer_service.write_advanced_configuration(
configuration_content=configuration_content)
return ApiResponse(success=True)
except ApplicationError as e:
raise HTTPException(status_code=400, detail=e.__str__())
except Exception as e:
raise HTTPException(status_code=500, detail=e.__str__())
async def start_eval(hyper_params: HyperParameterInformation,
background_tasks: BackgroundTasks):
try:
network_info: NetworkInformation = NetworkInformation(
network_architecture=hyper_params.network_architecture,
model_name=hyper_params.name)
background_tasks.add_task(
train_eval_continuously_manager.train_eval_continuously,
hyper_params=hyper_params)
background_tasks.add_task(export_manager.save_trained_model,
network_info=network_info)
return ApiResponse(success=True)
except ApplicationError as e:
print(e.__str__())
raise HTTPException(status_code=400, detail=e.__str__())
except Exception as e:
raise HTTPException(status_code=500, detail=e.__str__())
async def create_configuration(hyper_config_params: HyperParameterInformation):
try:
network_info: NetworkInformation = NetworkInformation(
model_name=hyper_config_params.name,
network_architecture=hyper_config_params.network_architecture)
configuration_file_content = configuration_service.get_config_file_content(
network_info=network_info)
configuration_obj = ConfigurationFactory().create_config_file(
network_info=network_info)
configuration_pipeline: Dict[
str, str] = configuration_obj.config_network(
config_file_content=configuration_file_content,
config_params=hyper_config_params)
configuration_writer_service.write_configuration(
configuration_pipeline=configuration_pipeline)
return ApiResponse(success=True)
except ApplicationError as e:
raise HTTPException(status_code=400, detail=e.__str__())
except Exception as e:
raise HTTPException(status_code=500, detail=e.__str__())