rai-predictive-training

rai-predictive-training is a skill for Claude Code from RelationalAI/rai-agent-skills. It costs 65 tokens per session (7,326 once invoked), scanned A, original, Apache-2.0.

A workflow for configuring, training, evaluating, and managing graph neural network models. These models learn from entities and the connections between them.

In plain words
What is it for?
It is for setting training options, fitting models, generating test predictions, checking results, debugging performance, and registering or loading trained models.
Why use it?
It keeps model training, prediction, evaluation, and saved-model handling in one process, while making clear that the underlying graph and features must be prepared elsewhere.

Skill for Claude Code

Written for Claude Code: shipped in a Claude Code plugin.

Part of the rai plugin — 12 skills shipped together

Good fit It is for setting training options, fitting models, generating test predictions, checking results, debugging performance, and registering or loading trained models.

Compare 6 skills from other repositories ↓
Install with agentmods
npx agentmods add skills/relationalai/rai-agent-skills/rai-predictive-training
Install

Getting it into your agent

One page per mod, every tool's command on it. A separate URL per tool would split the same page into five that compete with each other.

Any agent
npx skills add RelationalAI/rai-agent-skills --skill rai-predictive-training
Clone the repo
git clone --depth 1 https://github.com/RelationalAI/rai-agent-skills

Made for: Claude Code.

Or install rai, the plugin that ships this one along with the rest of its 12 skills.

Wrote this? Show the measurements

A badge with what this costs and how it scanned, read live from this page, so it follows the numbers instead of freezing them. Markdown for a README, HTML for a documentation site or a project page.

agentmods badge for rai-predictive-training

README.md
[![agentmods](https://agentmods.dev/badge/skills/relationalai/rai-agent-skills/rai-predictive-training/github.svg)](https://agentmods.dev/skills/relationalai/rai-agent-skills/rai-predictive-training)
Your own site
<a href="https://agentmods.dev/skills/relationalai/rai-agent-skills/rai-predictive-training"><img src="https://agentmods.dev/badge/skills/relationalai/rai-agent-skills/rai-predictive-training/github.svg" alt="Measured on agentmods" height="20"></a>

Or the 80×15 button, for a site that already has a row of RSS and ATOM ones. Only the verdict fits; the numbers stay here.

agentmods 80×15 button for rai-predictive-training

Your own site · 80×15
<a href="https://agentmods.dev/skills/relationalai/rai-agent-skills/rai-predictive-training"><img src="https://agentmods.dev/badge/skills/relationalai/rai-agent-skills/rai-predictive-training.svg" alt="Reviewed on agentmods" width="80" height="20"></a>
Per session 65 Skills are progressive disclosure: only the name and description are preloaded; the body loads when the skill is used.
When invoked 7,326 The whole file, excluding the scripts and references it only reads on demand.
Security scan A 0 findings. A grade says what 26 rules found in the file — not that it is safe.
Origin original No closer match found in the catalogue.
Token cost

What it costs to keep this loaded

Counted locally with the o200k_base tokenizer, which is exact for GPT models; Claude uses its own tokenizer and its counts differ. Treat this as one consistent yardstick across the catalogue rather than a bill. Prices are per million input tokens.

ModelPer sessionOnce invoked
Fable 5.1 $0.00065 $0.07326
Opus 5 $0.00032 $0.03663
Sonnet 5 $0.00013 $0.01465
Haiku 4.5 $0.00006 $0.00733

Measured 11d ago against content hash cd57962cefca, method: parsed. Prices are Anthropic first-party input rates as of 2026-09-11, from the pricing page.

Security

Grade A, and why

rai-predictive-training scanned grade A with 0 findings against 26 rules in 11 categories — prompt injection, anti-refusal, data exfiltration, privilege escalation, supply chain, agent snooping, system-prompt leakage, SSRF and excessive agency — measured 11d ago.

The scan reads SKILL.md. This mod also ships 4 executable files (examples/register_and_load.py, examples/train_link_prediction.py, examples/train_node_classification.py, …), listed below but not scanned — reading those needs a real analyzer, not pattern matching.

A static scan of the body, not an audit. Every finding is printed with the line that produced it so you can judge whether it matters here. A mod is markdown that instructs an agent; that is exactly why what it instructs is worth reading.

Nothing flagged

None of the 26 patterns this scan looks for appear in this file: no shell pipes, no recursive deletes, no credential paths, no hidden text, no instruction-override or anti-refusal phrasing, no agent-config snooping. That is not a guarantee, it is the absence of the things that are checkable.

plugins/rai/skills/rai-predictive-training/SKILL.md · 500 lines

How it starts

The opening of the file, as written. The whole thing — 500 lines — stays where its author put it; the contents beside it link to each section on GitHub.

Predictive Training

Early access. The RAI predictive reasoner (GNN) is in early access — APIs, engine requirements, and behavior may change. Confirm the latest surface with the RelationalAI team before production use.

Summary

What: Training, evaluation, and model management workflow for GNN pipelines.

When to use:

  • Configuring the GNN estimator and hyperparameters
  • Training models with fit()
  • Generating predictions on test data
  • Evaluating and debugging results
  • Registering or loading saved models

When NOT to use:

  • Defining concepts, loading data, building graphs -- see rai-predictive-modeling

Overview: 4 steps: configure GNN -> train -> predict/evaluate -> optional: register/load.

By user intent — sections to focus on:

  • Train + read validation metric → Quick Reference + GNN Constructor + gnn.fit()
    • predict + downstream rule / optimization → also Predictions + Using Predictions Downstream
    • register + reload across sessions → also Model Management

Quick Reference

Node Classification (minimal)

gnn = GNN(
    exp_database="DB", exp_schema="EXPERIMENTS",
    graph=gnn_graph, property_transformer=pt,
    train=Train, validation=Val,
    task_type="binary_classification", eval_metric="roc_auc",
    has_time_column=True, device="cuda", seed=42,
)
gnn.fit()
User.predictions = gnn.predictions(domain=Test)

Default Metrics

Task Type Suggested Metric
binary_classification roc_auc
multiclass_classification accuracy
multilabel_classification multilabel_auprc_macro
regression rmse
link_prediction link_prediction_precision@5
repeated_link_prediction link_prediction_precision@5

Prediction Attributes

Task Type Attributes
binary_classification, multiclass_classification, multilabel_classification .probs, .predicted_labels
regression .predicted_value
link_prediction, repeated_link_prediction .rank, .scores, .predicted_<target>

Read the full file on GitHub · 500 lines

Changes

What this file has done since we first saw it

Hashed on every crawl. A supply-chain change to an agent config is a question of when, not whether, so the history is kept rather than the latest state alone.

  1. 11d ago First seen · 500 lines · 65 tokens per session scan A cd57962cefca

Subscribe to this mod's changes

rai-predictive-training is a skill published in the GitHub repository RelationalAI/rai-agent-skills (4 stars, last pushed yesterday), licensed Apache-2.0. It adds 65 tokens to every session and 7,326 once invoked, about $0.0003 per session on Opus 5. A static security scan graded it A with 0 findings. No closer match exists in the catalogue, so it is treated as the original; first seen 2026-08-31.

Related

Other skills, from other repositories

agent-platform-rag-engine-management

Manage and query Agent Platform RAG Engine Corpora and retrieve grounded contexts using the Google GenAI SDK. Use when listing RAG corpora or files, inspecting a corpus, retrieving contexts, or generating content grounded in a RAG corpus. Do not use for standard database queries (use SQL/Spanner skills), Google…

google/skills · 85 tokens

agent-platform-model-registry

Agent Platform Model Registry Management. Use when you need to upload, list, describe, update, or delete machine learning models (and their versions) in the Agent Platform Model Registry. Don't use for model training, model deployment to endpoints, or managing non-Agent Platform models.

google/skills · 60 tokens

foundry-config-setup

Resolve missing setup caused by a hardcoded Foundry project endpoint or model in a sample. Use when a sample fails because it uses a placeholder/hardcoded projectendpoint (for example "https://your-project.services.ai.azure.com") or a hardcoded model instead of reading them from the environment.

microsoft/agent-framework · 65 tokens

google-cloud-solution-agentic-analytics-spark-knowledge-catalog

Discovers requirements and generates guidance to design and deploy a governed, secure agentic-analytics solution for data that's distributed across Google Cloud, other cloud providers, or on-premises. Data that's outside Google Cloud (such as data from Databricks, Snowflake, Salesforce, SAP, or Oracle systems) is…

google/skills · 138 tokens

training-check

Interactively monitor training metrics from the current Codex session, periodically checking WandB or fallback logs for NaN, divergence, plateaus, and broken runs.

wanshuiyin/Auto-claude-code-research-in-sleep · 35 tokens

nemo-automodel-launcher-config

Configure NeMo AutoModel job launches for interactive runs, Slurm clusters, and SkyPilot cloud execution.

NVIDIA/skills · 30 tokens