factor out nn_train.py into a separate python project to make it more manageable, add plotting and offline strength fit

1 files changed, 28 insertions, 0 deletions

diff --git a/train/pyproject.toml b/train/pyproject.toml
new file mode 100644
index 0000000..dcc77c6
--- /dev/null
+++ b/train/pyproject.toml
@@ -0,0 +1,28 @@
+[project]
+name = "ctak-train"
+version = "0.1.0"
+description = "Self-play training loop for the ctak engine"
+readme = "README.md"
+requires-python = ">=3.14"
+dependencies = [
+    "matplotlib>=3.11.2",
+    "numpy>=2.5.3",
+    "scipy>=1.18.1",
+    "torch>=2.14.0",
+    "tqdm>=4.70.1",
+]
+
+[project.scripts]
+nn-train = "ctak_train.nn_train:main"
+
+[build-system]
+requires = ["uv_build>=0.12.19,<0.13.0"]
+build-backend = "uv_build"
+
+[[tool.uv.index]]
+name = "pytorch-cpu"
+url = "https://download.pytorch.org/whl/cpu"
+explicit = true
+
+[tool.uv.sources]
+torch = { index = "pytorch-cpu" }