evaluation.py

from main import run

from src.costs import B1HomogeneityCost, B1HomogeneitySARCost
from src.data import Simulation
from src.utils import evaluate_coil_config

import numpy as np
import json

if __name__ == "__main__":
    # Load simulation data
    simulation = Simulation(path = "C:\\Users\\User\\Desktop\\Hackathon\\data\\simulations\\children_1_tubes_6_id_23713.h5",
                            coil_path = "C:\\Users\\User\\Desktop\\Hackathon\\data\\antenna\\antenna.h5")
    
    # Define cost function
    cost_function = B1HomogeneityCost()
    #cost_function = B1HomogeneitySARCost()

    
    # Run optimization
    best_coil_config = run(simulation=simulation, cost_function=cost_function)
    
    # Evaluate best coil configuration
    result = evaluate_coil_config(best_coil_config, simulation, cost_function)

    # Save results to JSON file
    with open("results.json", "w") as f:
        json.dump(result, f, indent=4)