Skip to content

GNNEstimator.fit()

Initializes a new training job on the dataset. Jobs initiated by the fit() method are of type train.

You can monitor the training process within Snowflake ML Experiments under the experiment_name, which uniquely identifies the dataset used for training. This name is automatically generated when creating the RelationalDataset in the format: dataset_name_task_type_task_name.

NameTypeDescriptionOptional
datasetRelationalDatasetThe dataset object used to train the model on its associated task.No

An instance of a JobMonitor object, to monitor the submitted job, check its status, track model progress, and retrieve metrics after training.

train_job = estimator.fit(dataset=dataset)