dict_dtype_slpy

dict_dtype = {
    "solute_or_solvent": "str",
    "metric_1": "float32",
    "metric_2": "float32",
    "epoch": "int32",
    "loss": "float32",
    "loss_ranking": "int32",
    "metric_1_ranking": "int32",
}