Model

A model

KEY TYPE Description
name str The user-friendly name for the model.
modelId str The unique identifier of the model.
modelConfigType str Name of the TrainingConfig class of the model_config.
modelPredictionConfig dict The prediction config options for the model.
createdAt str Date and time at which the model was created.
projectId str The project this model belongs to.
shared bool If model is shared to the Abacus.AI model showcase.
sharedAt str The date and time at which the model was shared to the model showcase
trainFunctionName str Name of the function found in the source code that will be executed to train the model. It is not executed when this function is run.
predictFunctionName str Name of the function found in the source code that will be executed run predictions through model. It is not executed when this function is run.
predictManyFunctionName str Name of the function found in the source code that will be executed to run batch predictions trhough the model.
initializeFunctionName str Name of the function found in the source code to initialize the trained model before using it to make predictions using the model
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 functions return value).
sourceCode str Python code used to make the model.
cpuSize str Cpu size specified for the python model training.
memory Int Memory in GB specified for the python model training.
trainingFeatureGroupIds List of Unique String Identifiers The unique identifiers of the feature groups used as the inputs to train this model on.
algorithmModelConfigs List[dict] List of algorithm specific training configs.
trainingVectorStoreVersions list The vector store version IDs used as inputs during training to create this ModelVersion.
documentRetrievers list List of document retrievers use to create this model.
documentRetrieverIds list List of document retriever IDs used to create this model.
isPythonModel bool If this model is handled as python model
defaultAlgorithm str If set, this algorithm will always be used when deploying the model regardless of the model metrics
customAlgorithmConfigs dict User-defined configs for each of the user-defined custom algorithm
restrictedAlgorithms dict User-selected algorithms to train.
useGpu bool If this model uses gpu.
notebookId str The notebook associated with this model.
trainingRequired bool If training is required to keep the model up-to-date.
latestModelVersion ModelVersion The latest model version.
location ModelLocation Location information for models that are imported.
refreshSchedules RefreshSchedule List of refresh schedules that indicate when the next model version will be trained
codeSource CodeSource If a python model, information on the source code
databaseConnector DatabaseConnector Database connector used by the model.
dataLlmFeatureGroups FeatureGroup List of feature groups used by the model for queries
modelConfig TrainingConfig The training config options used to train this model.
© Copyright