Skip to content

Predict All

Predict on an entire dataset using the specified model.

Parameters:

Name Type Description Default
report_name str

The name of the prediction report.

required
model_types list[int]

The model type identifiers to use for prediction.

required
model_id str

The ID of the model to use for prediction.

required
dataset_id str

The ID of the dataset to predict on.

required
actual_label_col_index int

The index of the actual label column.

required
input_text_col_index int

The index of the input text column.

required
document_files list[str] | None

List of paths to document files, if any.

None

Returns:

Name Type Description
dict Dict[str, Any]

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}")