types¶

Classes

`BaseData`([id])	A common dataclass for representing examples in a dataset.
`Example`([id, question, answer])	A common dataclass for representing examples in a dataset.
`GSM8KData`([id, question, answer, ...])	A dataclass for representing examples in the GSM8K dataset.
`HotPotQAData`([id, question, answer, ...])	A dataclass for representing examples in the HotPotQA dataset.
`TrecData`([id, question, class_name, class_index])	A dataclass for representing examples in the TREC dataset.

class BaseData(id: str = '30a4a2ae-8e0d-48b9-b648-09eab438b542')[source]¶

Bases: DataClass

A common dataclass for representing examples in a dataset.

id: str = '30a4a2ae-8e0d-48b9-b648-09eab438b542'¶

class Example(id: str = 'd6848ef8-7d79-4905-a100-6d52072ec398', question: str = None, answer: str = None)[source]¶

Bases: DataClass

A common dataclass for representing examples in a dataset.

id: str = 'd6848ef8-7d79-4905-a100-6d52072ec398'¶

question: str = None¶

answer: str = None¶

class GSM8KData(id: str = 'd6848ef8-7d79-4905-a100-6d52072ec398', question: str = None, answer: str = None, gold_reasoning: str = None, reasoning: str = None)[source]¶

Bases: Example

A dataclass for representing examples in the GSM8K dataset.

You can reset the output fields:

GSM8KData.set_output_fields(["answer"])

gold_reasoning: str = None¶

reasoning: str = None¶

class HotPotQAData(id: str = 'd6848ef8-7d79-4905-a100-6d52072ec398', question: str = None, answer: str = None, gold_titles: set = None, context: Dict[str, object] = None)[source]¶

Bases: Example

A dataclass for representing examples in the HotPotQA dataset.

gold_titles: set = None¶

context: Dict[str, object] = None¶

class TrecData(id: str = '30a4a2ae-8e0d-48b9-b648-09eab438b542', question: str = None, class_name: str = None, class_index: int = -1)[source]¶

Bases: BaseData

A dataclass for representing examples in the TREC dataset.

question: str = None¶

class_name: str = None¶

class_index: int = -1¶