Tensorboard Query

Tensorboard Query

An MCP server for querying and analyzing TensorBoard event files.

2
Stars
0
Forks
1
Releases

Overview

tb-query is a CLI tool and MCP server that lets AI agents and developers query and analyze TensorBoard event files without running a TensorBoard server. It reads events.out.tfevents.* files to extract scalar data, filter by tags and steps, compute statistics, and measure correlations between metrics. The MCP interface exposes tools for querying data, listing tags, locating event files, retrieving tag steps, computing tag statistics, and calculating correlations, enabling integration with AI coding assistants. It supports starting the MCP server with tb-query-mcp and the optional TB_QUERY_EVENTS_PATH environment variable to point to a default logs directory; an event_files resource lists available files. The project also provides a Python API (tb_query.core) and a variety of integration examples for Claude Desktop, Cline, Zed, and Continue. Error handling uses ValidationError for file access or parsing issues.

Details

Owner
Alir3z4
Language
Python
License
GNU General Public License v3.0
Updated
2025-12-07

Features

Query scalar data

Query scalar data with step and tag filtering from TensorBoard event files.

Find event files

Find all TensorBoard event files in a directory tree.

List available tags

List available scalar tags with optional filtering.

Calculate statistics

Calculate statistics (min, max, mean, std) for specific tags.

Compute correlations

Compute correlations between different scalar metrics.

CLI interface

Provide a CLI interface for convenient command-line usage.

MCP server integration

Offer an MCP server to enable AI coding assistants to interact with TensorBoard data.

Audience

AI agentsAllows AI agents to query and analyze TensorBoard events through MCP for training insights.
DevelopersEmbed TensorBoard data insights into apps and workflows via the MCP server.

Tags

TensorBoardTensorBoard eventsMCPModel Context Protocoltb-queryCLIevent filesscalarsstatisticscorrelationAI integrationPython APItools