Skip to main content
POST
/
v1
/
predict
import os, json, requests # Define your test dataset path test_path = "test.csv" # Get your API key from the environment api_key = os.getenv("PRIORLABS_API_KEY") headers = {"Authorization": f"Bearer {api_key}"} # Create prediction payload payload = { "task": "classification", "model_id": model_id, # Use model_id from your /v1/fit call } files = { "data": (None, json.dumps(payload), "application/json"), "file": (test_path, open(test_path, "rb")), } predict_response = requests.post( "https://api.priorlabs.ai/v1/predict", headers=headers, files=files, ) print("✅ Predictions:") print(json.dumps(predict_response.json(), indent=2))
{ "duration_seconds": 15, "prediction": [ [ 0.1, 0.9 ], [ 0.8, 0.2 ] ], "task": "classification", "params": { "average_before_softmax": false, "categorical_features_indices": null, "device": [ "cpu" ], "differentiable_input": false, "fit_mode": "fit_preprocessors", "ignore_pretraining_limits": true, "inference_config": null, "inference_precision": "auto", "memory_saving_mode": true, "model_path": "auto", "n_estimators": 8, "n_jobs": null, "n_preprocessing_jobs": 4, "random_state": 42, "softmax_temperature": 0.2 }, "used_credits": 10, "remaining_quota": 90 }

Authorizations

​
Authorization
string
header
required

Bearer token for authentication, obtained after signing up and generating an API key.

Body

multipart/form-data
​
data
string
required

A JSON string defining the prediction request parameters.

Required fields:

  • model_id (str) - Model ID from your previous /v1/fit call
  • task (str) - Task type: "classification" or "regression"

Optional Config Parameters:

  • systems (list[str]) - default: ["preprocessing", "text"]. The following preprocessing systems are supported:
    • ["preprocessing"] - Applies skrub preprocessing,
    • ["text"] - Adds text embeddings for text columns.
  • n_estimators (int) - Number of estimators in the ensemble (1-10)
  • model_path (str) - Model checkpoint path from HuggingFace
  • categorical_features_indices (List[int]) - Indices of categorical features
  • softmax_temperature (float) - Temperature for softmax scaling
  • average_before_softmax (bool) - Average before applying softmax
  • ignore_pretraining_limits (bool) - Ignore pretraining limits
  • inference_precision (str) - Inference precision ("float32", "float16", "auto")
  • random_state (int) - Random seed for reproducibility
  • balance_probabilities (bool) - Balance class probabilities

Optional Params (output configuration):

  • output_type (str) - Determines prediction output format
    • Classification: "probas" (default, probabilities) or "preds" (predictions)
    • Regression: "mean" (default, mean value) or "full" (includes quantiles, ei, pi)
​
file
file
required

CSV file containing the dataset to predict on.

Response

Prediction completed successfully — returns predicted values or probabilities depending on the task.

​
duration_seconds
integer
required

Time taken (in seconds) to complete the prediction.

​
prediction
required

The prediction output. Format depends on task and output_type:

  • Regression: List of floats (e.g., [1250.5, 3200.8, 980.2])
  • Classification with output_type: probas: List of lists (probabilities) (e.g., [[0.1, 0.9], [0.8, 0.2]])
  • Classification with output_type: preds: List of classes (e.g., ["class_0", "class_1"])
​
task
enum<string>
required

Specifies the type of task to perform — either classification or regression.

Available options:
classification,
regression
​
used_credits
integer
required

The number of credits consumed by this API call.

​
remaining_quota
integer
required

Your remaining credit balance after this request.

​
params
object

The inference parameters that were used during prediction.