From ffbe8e0b91713ae5d8e08e72d7e66b7731a6a72d Mon Sep 17 00:00:00 2001 From: samarthjain2023 Date: Sat, 27 Sep 2025 14:35:42 -0400 Subject: [PATCH] model train commands --- roadcast/command.txt | 4 ++++ 1 file changed, 4 insertions(+) create mode 100644 roadcast/command.txt diff --git a/roadcast/command.txt b/roadcast/command.txt new file mode 100644 index 0000000..b36db8e --- /dev/null +++ b/roadcast/command.txt @@ -0,0 +1,4 @@ +train the model: +python train.py data.csv --model-type mlp --generate-labels --label-method kmeans --n-buckets 50 --hidden-dims 512,256 --epochs 8 --batch-size 256 --feature-engineer --weight-decay 1e-5 --seed 42 + +python train.py data.csv --model-type mlp --generate-labels --label-method kmeans --n-buckets 50 --hidden-dims 1024,512 --epochs 12 --batch-size 256 --lr 1e-3 --lr-step-size 4 --lr-gamma 0.5 --feature-engineer --weight-decay 1e-5 --seed 42 \ No newline at end of file