sankalpsthakur/tiny-industrial-edge-onnx

Tiny Industrial Edge ONNX Classifier

A tiny scikit-learn LogisticRegression pipeline (StandardScaler + LR) trained on synthetic tabular features with industrial-energy-flavored names, exported to ONNX for CPU / edge demos.

Claim boundary

This model uses fully synthetic data. Feature names (specific_power, vibration_rms, steam_makeup, etc.) are illustrative only. Do not treat predictions as plant savings, control advice, or operational recommendations. Synthetic ≠ measured plant outcomes.

Features

Feature	Description (synthetic)
specific_power	Relative specific power
vibration_rms	Vibration RMS
steam_makeup	Steam makeup rate
condenser_approach	Condenser approach
cooling_tower_delta_t	Cooling tower delta-T
boiler_o2_pct	Boiler O₂ %

Label: intervene (binary)

Files

• model.onnx — ONNX graph (opset 12)
• latency.json — local CPU ORT benchmark
• train_export.py — reproducible train + export

Latency (CPU, onnxruntime)

From latency.json (1000 single-row inferences, CPUExecutionProvider, warm-up excluded):

Metric	Value
mean_ms	0.0046
p50_ms	0.0038
p95_ms	0.0090
provider	CPUExecutionProvider
n_inferences	1000

Numbers above match the committed latency.json from the export run (macOS ARM CPU).

Run with onnxruntime

pip install onnxruntime numpy
python - <<'PY'
import json
import numpy as np
import onnxruntime as ort

session = ort.InferenceSession("model.onnx", providers=["CPUExecutionProvider"])
name = session.get_inputs()[0].name
# order: specific_power, vibration_rms, steam_makeup,
#        condenser_approach, cooling_tower_delta_t, boiler_o2_pct
x = np.array([[0.9, 3.0, 5.0, 9.0, 11.0, 4.0]], dtype=np.float32)
label, probs = session.run(None, {name: x})
print("label", label, "probs", probs)
print(json.load(open("latency.json"))["mean_ms"], "ms mean")
PY

Reproduce export

python3 -m venv .venv-onnx
.venv-onnx/bin/pip install scikit-learn onnx onnxruntime skl2onnx numpy
.venv-onnx/bin/python train_export.py

Seed is fixed (SEED=42).

Links

• Transient Labs (Kaggle)
• Author profile: sankalpsthakur