Define output structure¶

In [7]:

from urbanworm import UrbanDataSet
from urbanworm.format_creation import create_format

from urbanworm import UrbanDataSet from urbanworm.format_creation import create_format

To structure the reponse from model, we can define a new Response class, which can hold a list of QnA class. In the QnA class, we can specify our expected outputs.

In [8]:

output_structure = create_format(
    {
        "question": (str, ...),
        "answer": (str, ...),
        "confidence": (float, ...),
    }
)

output_structure = create_format( { "question": (str, ...), "answer": (str, ...), "confidence": (float, ...), } )

In [5]:

data = UrbanDataSet(format=output_structure)
data.format

data = UrbanDataSet(format=output_structure) data.format

Out[5]:

urbanworm.format_creation.Response[QnA]