FastEHR.adapters.adapter

Classes

Adapter

Wrapper for converting FastEHR datasets to various supported frameworks

Module Contents

class FastEHR.adapters.adapter.Adapter(task: str, tokenizer, supervised: bool = False)

Wrapper for converting FastEHR datasets to various supported frameworks

Parameters

task: str

The downstream model format required. Supported formats include BEHRT.

tokenizer:

The FastEHR tokenizer

supervised: bool
Whether the FastEHR dataset is supervised or not. This is passed to

the adapter so we know if the last token in the context needs to have the SEP token added.

property tokenizer