88import igel
99import pandas as pd
1010from igel import Igel , metrics_dict
11- from igel .auto import IgelCNN
1211from igel .constants import Constants
1312from igel .servers import fastapi_server
1413from igel .utils import print_models_overview , show_model_info , tableize
@@ -73,29 +72,6 @@ def fit(data_path: str, yaml_path: str) -> None:
7372 Igel (cmd = "fit" , data_path = data_path , yaml_path = yaml_path )
7473
7574
76- @cli .command (context_settings = CONTEXT_SETTINGS )
77- @click .option (
78- "--data_path" , "-dp" , required = True , help = "Path to your training dataset"
79- )
80- @click .option (
81- "--task" ,
82- "-t" ,
83- required = False ,
84- help = "task you want to run. This refers to the goal you want to achieve (e.g ImageClassification)" ,
85- )
86- @click .option (
87- "--yaml_path" ,
88- "-yml" ,
89- required = False ,
90- help = "Path to your igel configuration file (yaml or json file)" ,
91- )
92- def auto_train (data_path : str , task : str , yaml_path : str ) -> None :
93- """
94- Automatically search for and train a suitable deep neural network for a task
95- """
96- IgelCNN (cmd = "train" , data_path = data_path , task = task , yaml_path = yaml_path )
97-
98-
9975@cli .command (context_settings = CONTEXT_SETTINGS )
10076@click .option (
10177 "--data_path" , "-dp" , required = True , help = "Path to your evaluation dataset"
@@ -107,17 +83,6 @@ def evaluate(data_path: str) -> None:
10783 Igel (cmd = "evaluate" , data_path = data_path )
10884
10985
110- @cli .command (context_settings = CONTEXT_SETTINGS )
111- @click .option (
112- "--data_path" , "-dp" , required = True , help = "Path to your evaluation dataset"
113- )
114- def auto_evaluate (data_path : str ) -> None :
115- """
116- Evaluate the performance of an existing machine learning model
117- """
118- IgelCNN (cmd = "evaluate" , data_path = data_path )
119-
120-
12186@cli .command (context_settings = CONTEXT_SETTINGS )
12287@click .option ("--data_path" , "-dp" , required = True , help = "Path to your dataset" )
12388def predict (data_path : str ) -> None :
@@ -127,15 +92,6 @@ def predict(data_path: str) -> None:
12792 Igel (cmd = "predict" , data_path = data_path )
12893
12994
130- @cli .command (context_settings = CONTEXT_SETTINGS )
131- @click .option ("--data_path" , "-dp" , required = True , help = "Path to your dataset" )
132- def auto_predict (data_path : str ) -> None :
133- """
134- Use an existing machine learning model to generate predictions
135- """
136- IgelCNN (cmd = "predict" , data_path = data_path )
137-
138-
13995@cli .command (context_settings = CONTEXT_SETTINGS )
14096@click .option (
14197 "--data_paths" ,
@@ -161,31 +117,6 @@ def experiment(data_paths: str, yaml_path: str) -> None:
161117 Igel (cmd = "predict" , data_path = pred_data_path )
162118
163119
164- @cli .command (context_settings = CONTEXT_SETTINGS )
165- @click .option (
166- "--data_paths" ,
167- "-DP" ,
168- required = True ,
169- help = "Path to your datasets as string separated by space" ,
170- )
171- @click .option (
172- "--yaml_path" ,
173- "-yml" ,
174- required = True ,
175- help = "Path to your igel configuration file (yaml or json file)" ,
176- )
177- def auto_experiment (data_paths : str , yaml_path : str ) -> None :
178- """
179- train, evaluate and use pre-trained model for predictions in one command
180- """
181- train_data_path , eval_data_path , pred_data_path = data_paths .strip ().split (
182- " "
183- )
184- IgelCNN (cmd = "train" , data_path = train_data_path , yaml_path = yaml_path )
185- IgelCNN (cmd = "evaluate" , data_path = eval_data_path )
186- IgelCNN (cmd = "predict" , data_path = pred_data_path )
187-
188-
189120@cli .command (context_settings = CONTEXT_SETTINGS )
190121@click .option (
191122 "--model_results_dir" ,
0 commit comments