Evaluation

KumoRFM provides an evaluation mode that automatically measures prediction quality by performing a train/test split on context examples and computing relevant metrics.

Running an Evaluation

Use KumoRFM.evaluate() with the same PQL syntax as `KumoRFM.predict()`:

1
metrics = model.evaluate(
2
    "PREDICT COUNT(orders.*, 0, 30, days) > 0 FOR users.user_id=1",
3
    run_mode="fast",
4
)
5
print(metrics)

The evaluation collects context examples, splits them into in-context (training) and test sets, generates predictions for the test set, and computes metrics comparing predictions to actual outcomes.

You can also use the EVALUATE keyword in the query string directly:

1
metrics = model.evaluate(
2
    "EVALUATE PREDICT COUNT(orders.*, 0, 30, days) FOR users.user_id=1"
3
)

Available Metrics

The metrics returned depend on the detected task type:

You can specify which metrics to compute:

1
metrics = model.evaluate(
2
    "PREDICT SUM(orders.price, 0, 30, days) FOR items.item_id=42",
3
    metrics=["mae", "rmse", "r2"],
4
)

Evaluation Parameters

The KumoRFM.evaluate() method accepts the same parameters as KumoRFM.predict(), plus:

• metrics: A list of metric names to compute. If not specified, all applicable metrics for the task type are computed.

The run_mode, anchor_time, num_hops, and other parameters work identically to KumoRFM.predict(). See configuration for details on run modes.

Evaluation with TaskTable

For advanced use cases, you can construct a TaskTable explicitly and use `KumoRFM.evaluate_task()`:

1
from kumoai.rfm import TaskTable
2
3
task = TaskTable(
4
    task_type="binary_classification",
5
    context_df=context_dataframe,
6
    pred_df=prediction_dataframe,
7
    entity_table_name="users",
8
    entity_column="user_id",
9
    target_column="target",
10
    time_column="timestamp",
11
)
12
13
metrics = model.evaluate_task(task)

This gives you full control over the train/test split and context construction.

Interpreting Results

The evaluation returns a pandas.DataFrame with metric and value columns:

1
>>> metrics = model.evaluate(query)
2
>>> print(metrics)
3
  metric  value
4
0    mae   12.5
5
1   rmse   15.3
6
2     r2   0.82

Higher values are better for r2, acc, auroc, auprc, ap, precision, recall, and f1. Lower values are better for mae, mape, mse, rmse, smape.