Skip to content

Predict All

Predict on an entire dataset using the specified model.

Parameters:

Name Type Description Default
model_id str

The ID of the model to use for prediction.

required
datasetId str

The ID of the dataset to predict on.

required
input_text_col_index int

The index of the input text column.

required
document_files list

List of paths to document files, if any.

None

Returns:

Name Type Description
dict

A dictionary containing the predictions.

Sample Usage

report_name = "batch_prediction_report"
dataset_id = "your_dataset_id"
actual_label_col_index = 1

predictions = anote.predictAll(
    report_name=report_name,
    model_types=[model_type],
    model_id=model_id,
    dataset_id=dataset_id,
    actual_label_col_index=actual_label_col_index,
    input_text_col_index=input_text_col_index
)
print(f"Batch Predictions: {predictions}")