Skip to content

Training the Regressor

Four steps, in order. The first is a long multi-GPU job, so read the constants at the top of collect_data.py before starting it. With a collected coverage dataset, start at train.

python -m intelx.cli.collect_data    # coverage data, over your maps
<CHUNKING_SCRIPT>                    # the collected shows, as training chunks
python -m intelx.cli.train           # the U-Net
python -m intelx.cli.evaluate        # what it predicts, side by side

collect_data runs the 18 training maps listed in data/train_metadata.csv, each with the number of shows Table 1 of the paper gives it, in proportion to its complexity score. It reads the maps from train/ under the maps root, samples spawns, runs the exploration shows that record how much new area each goal would gain, and writes the result under INTELX_DATASET_ROOT. intelx.data.driver does one map at a time and is what the controller drives. <CHUNKING_SCRIPT> packs the collected shows into train_chunks/. train reads those chunks, at batch size 32 for up to 300 epochs, keeps the checkpoint with the best validation R², and writes each run, with the settings it used, to weights/runs/<name>. evaluate loads a checkpoint and writes prediction overlays and inference timings to results/evaluation_large.

Everything each step reads and writes is a class attribute at the top of its file. None of the three takes command-line arguments.