def main(dataset: str, pipeline: str):
    models = ModelService()
    query = {'type': "FEATURES"}
    if dataset != 'all':
        query['name'] = dataset
    items = models.create_classification_models(query=query, pipeline=pipeline)
    print(items)
Example #2
0
def model_index(query: Optional[dict] = Body(...),
                service: ModelService = Depends(ModelService)):
    return service.create_classification_models(query)