tao-mine-nearest-neighbors

tao-mine-nearest-neighbors is a skill for Claude Code from NVIDIA-TAO/tao-skill-bank. It costs 39 tokens per session (1,490 once invoked), scanned A, original, Apache-2.0.

A nearest-neighbor search workflow for finding source data samples that are most similar to target samples. It works with parquet files containing numerical representations of data, called embeddings.

In plain words
What is it for?
Use it to select similar source samples for data mining, analysis, or later training workflows.
Why use it?
It removes the need to manually compare large sets of samples. It does not create embeddings, so those files must already exist.

Skill for Claude Code

Written for Claude Code: allowed-tools in frontmatter.

Part of the tao-skills plugin — 76 skills shipped together , and of tao-skill-bank

Good fit Use it to select similar source samples for data mining, analysis, or later training workflows.

Compare 6 skills from other repositories ↓
Install with agentmods
npx agentmods add skills/nvidia-tao/tao-skill-bank/tao-mine-nearest-neighbors
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 NVIDIA-TAO/tao-skill-bank --skill tao-mine-nearest-neighbors
Clone the repo
git clone --depth 1 https://github.com/NVIDIA-TAO/tao-skill-bank

Made for: Claude Code.

Or install tao-skills, the plugin that ships this one along with the rest of its 76 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 tao-mine-nearest-neighbors

README.md
[![agentmods](https://agentmods.dev/badge/skills/nvidia-tao/tao-skill-bank/tao-mine-nearest-neighbors/github.svg)](https://agentmods.dev/skills/nvidia-tao/tao-skill-bank/tao-mine-nearest-neighbors)
Your own site
<a href="https://agentmods.dev/skills/nvidia-tao/tao-skill-bank/tao-mine-nearest-neighbors"><img src="https://agentmods.dev/badge/skills/nvidia-tao/tao-skill-bank/tao-mine-nearest-neighbors/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 tao-mine-nearest-neighbors

Your own site · 80×15
<a href="https://agentmods.dev/skills/nvidia-tao/tao-skill-bank/tao-mine-nearest-neighbors"><img src="https://agentmods.dev/badge/skills/nvidia-tao/tao-skill-bank/tao-mine-nearest-neighbors.svg" alt="Reviewed on agentmods" width="80" height="20"></a>
Per session 39 Skills are progressive disclosure: only the name and description are preloaded; the body loads when the skill is used.
When invoked 1,490 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. Third-party audits
  • NVIDIA SkillSpector warn 7 Sept 2026
SkillSpector: 5 findings, up to high

These are SkillSpector’s own severities. On a checked sample its high-severity flags on skills were ~96% false positives — a documented command, a public API, a “never do X” rule — so we show them as a caution to read, not a verdict. Why →

  • high YARA Match · line 30
    YARA rule matched a hack tool or exploit indicator (offensive tools, reconnaissance, privilege escalation, or exploit frameworks).
    Fix: Remove offensive tool references and exploit code. Legitimate agent skills should not contain penetration testing tools, exploit frameworks, or reconnaissance utilities.
  • high Privilege Escalation · line 73
    Potential security issue detected. Manual review is recommended.
    Fix: Review the flagged content for security risks. Ensure no credentials, secrets, or sensitive data are exposed.
  • high Privilege Escalation · line 82
    Code accesses credential files (SSH keys, AWS credentials, etc.). This could indicate credential theft attempts.
    Fix: Remove references to credential paths. Use environment variables or secrets managers. For docs, use placeholder paths (e.g., /path/to/config). Never load .env or token files in production code paths.
  • medium MCP Rug Pull · line 73
    Docker image references without a specific tag (:latest is implicit) or digest (@sha256:...) can be silently replaced by a malicious image.
    Fix: Pin the image: image:tag or image@sha256:abc123
  • medium MCP Rug Pull · line 117
    Docker image references without a specific tag (:latest is implicit) or digest (@sha256:...) can be silently replaced by a malicious image.
    Fix: Pin the image: image:tag or image@sha256:abc123
How audits are shown
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.00039 $0.01490
Opus 5 $0.00019 $0.00745
Sonnet 5 $0.00008 $0.00298
Haiku 4.5 $0.00004 $0.00149

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

Security

Grade A, and why

tao-mine-nearest-neighbors 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 13d ago.

The scan reads SKILL.md. This mod also ships 2 executable files (scripts/prepare_nearest_neighbors_spec.py, scripts/verify_nearest_neighbors_spec.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.

skills/data/tao-mine-nearest-neighbors/SKILL.md · 149 lines

How it starts

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

TAO Mine Nearest Neighbors

Use this skill to run TAO Data Services TMM nearest-neighbor mining. The skill consumes embedding parquets and writes a mined source-sample parquet plus a mining summary. It does not compute embeddings; upstream steps must produce the source and target embedding parquets first.

The container entrypoint is:

tmm nearest_neighbors -e /absolute/path/to/nearest_neighbors.yaml

TAO Data Services requires -e/--experiment_spec_file. The tmm console script converts that YAML into Hydra --config-path and --config-name arguments internally.

Inputs

The user can provide either an existing nearest-neighbors YAML spec or the fields needed to generate one.

Required spec fields:

Field Meaning
source_parquet Absolute path to the candidate/source embeddings parquet.
target_parquet Absolute path to the target/query embeddings parquet.
output_parquet Absolute path where TAO Data Services should write mined source filepaths.

Common optional fields:

Field Default Meaning
topn 5 Number of nearest source samples to retrieve per target sample.
knn_metric cosine One of cosine, euclidean, or manhattan.
source_embed_column_name embedding Embedding column in source_parquet.
target_embed_column_name embedding Embedding column in target_parquet.
filter_by_label "false" String flag. When "true", TAO DS filters neighbors by matching label columns when both parquets provide labels.
distance_threshold -1.0 Maximum distance to keep. Negative disables thresholding.

Both input parquets must contain a filepath column and a list-like embedding column. If filter_by_label is "true", both parquets should also contain label.

The default template is assets/default_nearest_neighbors.yaml.

Quick Start

Run from the tao-skill-bank repo root. Resolve the pinned TAO Data Services image from versions.yaml, verify the spec, mount the run root with identical host/container paths, and stream the Docker logs.

Read the full file on GitHub · 149 lines

Files

What ships with it

5 files beside SKILL.md in the same directory: the scripts, references and assets a skill reads on demand. Not counted in the per-session cost; read them before you install if any of them is executable.

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. 13d ago First seen · 149 lines · 39 tokens per session scan A 7d3617a5d687

Subscribe to this mod's changes

tao-mine-nearest-neighbors is a skill published in the GitHub repository NVIDIA-TAO/tao-skill-bank (88 stars, last pushed today), licensed Apache-2.0. It adds 39 tokens to every session and 1,490 once invoked, about $0.0002 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-30.

Related

Other skills, from other repositories

arrowspace

Spectral vector search using graph Laplacian eigenstructure. Use when cosine/L2 similarity misses latent structure in your embeddings.

sickn33/agentic-awesome-skills · 28 tokens

docetl

Build and run LLM-powered data processing pipelines with DocETL. Use when users say "docetl", want to analyze unstructured data, process documents, extract information, or run ETL tasks on text. Helps with data collection, pipeline creation, execution, and optimization.

ucbepic/docetl · 59 tokens

senior-data-engineer

World-class data engineering skill for building scalable data pipelines, ETL/ELT systems, real-time streaming, and data infrastructure. Expertise in Python, SQL, Spark, Airflow, dbt, Kafka, Flink, Kinesis, and modern data stack. Includes data modeling, pipeline orchestration, data quality, streaming quality…

benchflow-ai/skillsbench · 100 tokens

malloy-model

Build Malloy semantic models with base source and joined source files. Use when creating or modifying .malloy files, user asks to "create a malloy model", "add dimensions", "add measures", "create a source", or any Malloy model authoring task.

malloydata/publisher · 59 tokens

malloy-model-as-you-go

After answering a data question, write down what the answer assumed so the next reader can trust the number. A field with a.

malloydata/publisher · 32 tokens

malloy-define

Propose a source plan and field definitions for a Malloy semantic model. Covers picking which sources to model and at what grain, then proposing the specific renames, dimensions, and measures per source, every proposal backed by querying the data.

malloydata/publisher · 53 tokens