1 line
15 KiB
JSON
1 line
15 KiB
JSON
{"nodes": [{"id": "modules_bot_python_src_train_py", "label": "train.py", "file_type": "code", "source_file": "modules/bot/python/src/train.py", "source_location": "L1"}, {"id": "train_nnuedataset", "label": "NNUEDataset", "file_type": "code", "source_file": "modules/bot/python/src/train.py", "source_location": "L17"}, {"id": "dataset", "label": "Dataset", "file_type": "code", "source_file": "", "source_location": ""}, {"id": "train_nnuedataset_init", "label": ".__init__()", "file_type": "code", "source_file": "modules/bot/python/src/train.py", "source_location": "L20"}, {"id": "train_nnuedataset_len", "label": ".__len__()", "file_type": "code", "source_file": "modules/bot/python/src/train.py", "source_location": "L48"}, {"id": "train_nnuedataset_getitem", "label": ".__getitem__()", "file_type": "code", "source_file": "modules/bot/python/src/train.py", "source_location": "L51"}, {"id": "train_fen_to_features", "label": "fen_to_features()", "file_type": "code", "source_file": "modules/bot/python/src/train.py", "source_location": "L64"}, {"id": "train_nnue", "label": "NNUE", "file_type": "code", "source_file": "modules/bot/python/src/train.py", "source_location": "L89"}, {"id": "train_nnue_init", "label": ".__init__()", "file_type": "code", "source_file": "modules/bot/python/src/train.py", "source_location": "L92"}, {"id": "train_nnue_forward", "label": ".forward()", "file_type": "code", "source_file": "modules/bot/python/src/train.py", "source_location": "L112"}, {"id": "train_find_next_version", "label": "find_next_version()", "file_type": "code", "source_file": "modules/bot/python/src/train.py", "source_location": "L119"}, {"id": "train_save_metadata", "label": "save_metadata()", "file_type": "code", "source_file": "modules/bot/python/src/train.py", "source_location": "L141"}, {"id": "train_setup_training", "label": "_setup_training()", "file_type": "code", "source_file": "modules/bot/python/src/train.py", "source_location": "L155"}, {"id": "train_run_training_season", "label": "_run_training_season()", "file_type": "code", "source_file": "modules/bot/python/src/train.py", "source_location": "L218"}, {"id": "train_save_versioned_model", "label": "_save_versioned_model()", "file_type": "code", "source_file": "modules/bot/python/src/train.py", "source_location": "L329"}, {"id": "train_train_nnue", "label": "train_nnue()", "file_type": "code", "source_file": "modules/bot/python/src/train.py", "source_location": "L370"}, {"id": "train_burst_train", "label": "burst_train()", "file_type": "code", "source_file": "modules/bot/python/src/train.py", "source_location": "L448"}, {"id": "train_rationale_18", "label": "Dataset of chess positions with evaluations.", "file_type": "rationale", "source_file": "modules/bot/python/src/train.py", "source_location": "L18"}, {"id": "train_rationale_65", "label": "Convert FEN to 768-dimensional binary feature vector.", "file_type": "rationale", "source_file": "modules/bot/python/src/train.py", "source_location": "L65"}, {"id": "train_rationale_90", "label": "NNUE neural network architecture: 768\u21921536\u21921024\u2192512\u2192256\u21921 with dropout regulariz", "file_type": "rationale", "source_file": "modules/bot/python/src/train.py", "source_location": "L90"}, {"id": "train_rationale_120", "label": "Find the next version number for model versioning. Looks for nnue_weights_v", "file_type": "rationale", "source_file": "modules/bot/python/src/train.py", "source_location": "L120"}, {"id": "train_rationale_142", "label": "Save training metadata alongside the weights file. Args: weights_fi", "file_type": "rationale", "source_file": "modules/bot/python/src/train.py", "source_location": "L142"}, {"id": "train_rationale_156", "label": "Set up device, dataset, and data loaders. Returns: (device, dataset", "file_type": "rationale", "source_file": "modules/bot/python/src/train.py", "source_location": "L156"}, {"id": "train_rationale_225", "label": "Run one training season until epoch limit, early stopping, or deadline. Arg", "file_type": "rationale", "source_file": "modules/bot/python/src/train.py", "source_location": "L225"}, {"id": "train_rationale_332", "label": "Save the best model with optional versioning and metadata.", "file_type": "rationale", "source_file": "modules/bot/python/src/train.py", "source_location": "L332"}, {"id": "train_rationale_371", "label": "Train the NNUE model with GPU optimizations and automatic mixed precision.", "file_type": "rationale", "source_file": "modules/bot/python/src/train.py", "source_location": "L371"}, {"id": "train_rationale_453", "label": "Train in burst mode: repeatedly restart from the best checkpoint until the time", "file_type": "rationale", "source_file": "modules/bot/python/src/train.py", "source_location": "L453"}], "edges": [{"source": "modules_bot_python_src_train_py", "target": "json", "relation": "imports", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L4", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "torch", "relation": "imports", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L5", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "torch_nn", "relation": "imports", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L6", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "torch_optim", "relation": "imports", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L7", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "torch_utils_data", "relation": "imports_from", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L8", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "sys", "relation": "imports", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L9", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "pathlib", "relation": "imports_from", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L10", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "tqdm", "relation": "imports_from", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L11", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "chess", "relation": "imports", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L12", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "datetime", "relation": "imports_from", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L13", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "re", "relation": "imports", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L14", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "numpy", "relation": "imports", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L15", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "train_nnuedataset", "relation": "contains", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L17", "weight": 1.0}, {"source": "train_nnuedataset", "target": "dataset", "relation": "inherits", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L17", "weight": 1.0}, {"source": "train_nnuedataset", "target": "train_nnuedataset_init", "relation": "method", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L20", "weight": 1.0}, {"source": "train_nnuedataset", "target": "train_nnuedataset_len", "relation": "method", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L48", "weight": 1.0}, {"source": "train_nnuedataset", "target": "train_nnuedataset_getitem", "relation": "method", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L51", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "train_fen_to_features", "relation": "contains", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L64", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "train_nnue", "relation": "contains", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L89", "weight": 1.0}, {"source": "train_nnue", "target": "train_nnue_init", "relation": "method", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L92", "weight": 1.0}, {"source": "train_nnue", "target": "train_nnue_forward", "relation": "method", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L112", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "train_find_next_version", "relation": "contains", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L119", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "train_save_metadata", "relation": "contains", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L141", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "train_setup_training", "relation": "contains", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L155", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "train_run_training_season", "relation": "contains", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L218", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "train_save_versioned_model", "relation": "contains", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L329", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "train_train_nnue", "relation": "contains", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L370", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "train_burst_train", "relation": "contains", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L448", "weight": 1.0}, {"source": "modules_bot_python_src_train_py", "target": "argparse", "relation": "imports", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L571", "weight": 1.0}, {"source": "train_nnuedataset_getitem", "target": "train_fen_to_features", "relation": "calls", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L54", "weight": 1.0}, {"source": "train_setup_training", "target": "train_nnuedataset", "relation": "calls", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L172", "weight": 1.0}, {"source": "train_save_versioned_model", "target": "train_find_next_version", "relation": "calls", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L338", "weight": 1.0}, {"source": "train_save_versioned_model", "target": "train_save_metadata", "relation": "calls", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L364", "weight": 1.0}, {"source": "train_train_nnue", "target": "train_setup_training", "relation": "calls", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L387", "weight": 1.0}, {"source": "train_train_nnue", "target": "train_nnue", "relation": "calls", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L389", "weight": 1.0}, {"source": "train_train_nnue", "target": "train_run_training_season", "relation": "calls", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L427", "weight": 1.0}, {"source": "train_train_nnue", "target": "train_save_versioned_model", "relation": "calls", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L440", "weight": 1.0}, {"source": "train_burst_train", "target": "train_setup_training", "relation": "calls", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L476", "weight": 1.0}, {"source": "train_burst_train", "target": "train_nnue", "relation": "calls", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L478", "weight": 1.0}, {"source": "train_burst_train", "target": "train_run_training_season", "relation": "calls", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L522", "weight": 1.0}, {"source": "train_burst_train", "target": "train_save_versioned_model", "relation": "calls", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L554", "weight": 1.0}, {"source": "train_rationale_18", "target": "train_nnuedataset", "relation": "rationale_for", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L18", "weight": 1.0}, {"source": "train_rationale_65", "target": "train_fen_to_features", "relation": "rationale_for", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L65", "weight": 1.0}, {"source": "train_rationale_90", "target": "train_nnue", "relation": "rationale_for", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L90", "weight": 1.0}, {"source": "train_rationale_120", "target": "train_find_next_version", "relation": "rationale_for", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L120", "weight": 1.0}, {"source": "train_rationale_142", "target": "train_save_metadata", "relation": "rationale_for", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L142", "weight": 1.0}, {"source": "train_rationale_156", "target": "train_setup_training", "relation": "rationale_for", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L156", "weight": 1.0}, {"source": "train_rationale_225", "target": "train_run_training_season", "relation": "rationale_for", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L225", "weight": 1.0}, {"source": "train_rationale_332", "target": "train_save_versioned_model", "relation": "rationale_for", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L332", "weight": 1.0}, {"source": "train_rationale_371", "target": "train_train_nnue", "relation": "rationale_for", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L371", "weight": 1.0}, {"source": "train_rationale_453", "target": "train_burst_train", "relation": "rationale_for", "confidence": "EXTRACTED", "source_file": "modules/bot/python/src/train.py", "source_location": "L453", "weight": 1.0}]} |