REQUIRED |
KEY |
TYPE |
DESCRIPTION |
Yes |
modelId |
str |
The unique ID associated with the Python model to be changed.
|
No |
trainFunctionName |
str |
Name of the function found in the train module that will be executed to train the model. It is not executed when this function is run.
|
No |
predictFunctionName |
str |
Name of the function found in the predict module that will be executed to run predictions through the model. It is not executed when this function is run.
|
No |
predictManyFunctionName |
str |
Name of the function found in the predict module that will be executed to run batch predictions through the model. It is not executed when this function is run.
|
No |
trainModuleName |
str |
Full path of the module that contains the train function from the root of the zip.
|
No |
predictModuleName |
str |
Full path of the module that contains the predict function from the root of the zip.
|
No |
trainingInputTables |
list |
List of feature groups that are supplied to the train function as parameters. Each of the parameters are materialized Dataframes (same type as the function's return value).
|
No |
cpuSize |
str |
Size of the CPU for the model training function.
|
No |
memory |
int |
Memory (in GB) for the model training function.
|
No |
packageRequirements |
list |
List of package requirement strings. For example: ['numpy==1.2.3', 'pandas>=1.4.0'].
|
No |
useGpu |
bool |
Whether this model needs gpu
|
Note: The arguments for the API methods follow camelCase but for Python SDK underscore_case is followed.