mattertune.data.json_data

Classes

JSONDataset(config)

JSONDatasetConfig(*[, type])

class mattertune.data.json_data.JSONDatasetConfig(*, type='json', src, tasks)[source]
Parameters:

  • type (Literal['json'])

  • src (str | Path)

  • tasks (dict[str, str])

type: Literal['json']

Discriminator for the JSON dataset.

src: str | Path

The path to the JSON dataset.

tasks: dict[str, str]

Attributes in the JSON file that correspond to the tasks to be predicted.

create_dataset()[source]
class mattertune.data.json_data.JSONDataset(config)[source]
Parameters:

config (JSONDatasetConfig)

__init__(config)[source]
Parameters:

config (JSONDatasetConfig)