Update README.md

README.md

@@ -1,41 +1,46 @@
-# SolveQ
-
-Simple neural network that learns `y = 2x + 3`. Built with PyTorch.
-
-## What it does
-
-Trains a small network to approximate a linear function. The model has one hidden layer and learns from synthetic data.
-
-## Setup
-
-```bash
-pip install -r requirements.txt
-```
-
-## Running
-
-Train the model:
-
-```bash
-python src/train.py
-```
-
-This generates 200 samples, trains for 1000 epochs, shows the loss plot, and saves to `models/model.pt`.
-
-![Training Loss](models/loss.png)
-
-Make predictions:
-
-```bash
-python src/predict.py
-```
-
-Predicts `x = 4.0` by default. To test other values, edit the `predict()` call in `predict.py`.
-
-## Model
-
-- Input: 1 neuron
-- Hidden: 16 neurons (ReLU)
-- Output: 1 neuron
-
-Uses Adam optimizer (lr=0.01) and MSE loss.
+---
+license: mit
+metrics:
+- accuracy
+---
+# SolveQ
+
+Simple neural network that learns `y = 2x + 3`. Built with PyTorch.
+
+## What it does
+
+Trains a small network to approximate a linear function. The model has one hidden layer and learns from synthetic data.
+
+## Setup
+
+```bash
+pip install -r requirements.txt
+```
+
+## Running
+
+Train the model:
+
+```bash
+python src/train.py
+```
+
+This generates 200 samples, trains for 1000 epochs, shows the loss plot, and saves to `models/model.pt`.
+
+![Training Loss](models/loss.png)
+
+Make predictions:
+
+```bash
+python src/predict.py
+```
+
+Predicts `x = 4.0` by default. To test other values, edit the `predict()` call in `predict.py`.
+
+## Model
+
+- Input: 1 neuron
+- Hidden: 16 neurons (ReLU)
+- Output: 1 neuron
+
+Uses Adam optimizer (lr=0.01) and MSE loss.