DeepChess-AI/setup.py at master · shafitek/DeepChess-AI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import os
import json

from meta import *

def main():

    print("-------------------")
    print("| DeepChess Setup |")
    print("-------------------\n")

    print("Base Directory: {}\n".format(BASE_DIR))

    if os.path.exists(META_FILE):
        print("Setup already exectued.")
        print("---")
        return 1

    meta_def = {
            "dataset_url": "http://ccrl.chessdom.com/ccrl/4040/CCRL-4040.[1183411].pgn.7z",
            "dataset_downloaded": "False",
            "dataset_decompressed": "False",
            "dataset_offsets_calculated": "False",
            "dataset_generated": "False",
            "time_to_calc_dataset_offset": -1,
            "time_to_generate_dataset": -1,
            "compressed_dataset_path": "",
            "dataset_white_path": "",
            "dataset_white_offset_path": "",
            "dataset_black_path": "",
            "dataset_black_offset_path": "",
            "pos2vec_trained": "False",
            "pos2vec_weights_path": "",
            "deepchess_trained": "False",
            "deepchess_model_path": "",
            "deepchess_epochs_trained": 0,
            "max_num_white_positions_train": 2100000,
            "max_num_black_positions_train": 1600000,
            "num_white_positions_train": 0,
            "num_black_positions_train": 0,
            "num_white_positions_test": 100000,
            "num_black_positions_test": 100000
        }

    with open(META_FILE, 'w') as f:
        json.dump(meta_def, f, indent=4)

    print("meta.json created at:", META_FILE)
    print("Setup complete.")
    print("---")


if __name__ == "__main__":
    main()