Skip to content

graphchem.datasets.load_ysi

Loads yield sooting index data.

Returns

Tuple[List[str], torch.Tensor] A tuple containing two elements: - List[str]: A list of SMILES strings. - torch.Tensor: A tensor of YSI values with dtype float32.

Source code in graphchem/datasets/sets.py 
 96
 97
 98
 99
100
101
102
103
104
105
106
107
def load_ysi() -> Tuple[List[str], torch.Tensor]:
    """
    Loads yield sooting index data.

    Returns
    -------
    Tuple[List[str], torch.Tensor]
        A tuple containing two elements:
        - List[str]: A list of SMILES strings.
        - torch.Tensor: A tensor of YSI values with dtype float32.
    """
    return _load_set("ysi")