PsychiatryNLPKit.model.HFMaskFillingModel

class PsychiatryNLPKit.model.HFMaskFillingModel(name, key=None)[source]

Marker for masked language models.

__init__(name, key=None)[source]

Methods

`__init__`(name[, key])
`load`()
`unload`()

Attributes

`model`
`tokenizer`

__init__(name, key=None)[source]