ComputerAssignments/CA6_systematic_generalization/setup_structure.py

import os

def create_project_structure():

    base_dir = "/Users/tahamajs/Documents/uni/Sysmem2_in_AI/CAs/CA6"


    directories = [
        "src",
        "src/core",
        "src/models",
        "src/datasets",
        "src/evaluation",
        "src/experiments",
        "src/visualization",
        "configs",
        "data",
        "results",
        "results/experiments",
        "results/plots",
        "results/models",
        "logs",
        "tests",
        "scripts",
    ]

    for directory in directories:
        dir_path = os.path.join(base_dir, directory)
        os.makedirs(dir_path, exist_ok=True)
        print(f"Created directory: {dir_path}")


    init_files = [
        "src/__init__.py",
        "src/core/__init__.py",
        "src/models/__init__.py",
        "src/datasets/__init__.py",
        "src/evaluation/__init__.py",
        "src/experiments/__init__.py",
        "src/visualization/__init__.py",
        "tests/__init__.py",
    ]

    for init_file in init_files:
        file_path = os.path.join(base_dir, init_file)
        with open(file_path, "w") as f:
            f.write("")
        print(f"Created file: {file_path}")

if __name__ == "__main__":
    create_project_structure()