Skip to content

graphchem.datasets.load_cn

Loads cetane number data.

Returns

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

Source code in graphchem/datasets/sets.py 
40
41
42
43
44
45
46
47
48
49
50
51
def load_cn() -> Tuple[List[str], torch.Tensor]:
    """
    Loads cetane number data.

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