ml-trainer-mcp 0.2.0

An MCP server that trains and evaluates machine learning models through MCP tools

ML Trainer MCP

An MCP server that trains and evaluates machine learning models through MCP tools.

Quickstart

uv pip install -e .
uv run ml-trainer-mcp

Or test interactively with the MCP Inspector:

mcp dev src/ml_trainer_mcp/server.py

Tools

Tool	Description
load_csv(path)	Load a CSV file as the active dataset
dataset_info()	Shape, dtypes, and summary stats
list_columns()	List column names
missing_values()	Report missing values per column
train_linear_regression(target)	Train a linear regression model
train_random_forest(target, ...)	Train a random forest regressor
evaluate_model(test_size)	Evaluate the trained model on a held-out split
predict(input_data)	Predict on a single new row
feature_importance()	Feature importances / coefficients