Model-Specific Parameter Definition Template
Below is a template for defining model-specific parameters.
If no params are required by the model, then it should be an empty list.
The file should be saved as model_params_def.py
so that each list is imported correctly by the model scripts.
For more information on setting model-specific parameters see Creating Model-Specific Parameters.
from improvelib.utils import str2bool
preprocess_params = [
{"name": "name1",
"type": str2bool,
"default": True,
"help": "Help goes here",
},
{"name": "name2",
"type": str,
"default": "text",
"help": "Help goes here.",
},
{"name": "name3",
"type": int,
"default": 2,
"help": "Help goes here.",
},
]
train_params = [
{"name": "name4",
"type": float,
"default": .02,
"help": "Help goes here."
},
]
infer_params = []