INNER CODE UNIT · Python

__init__

hooshvare/parsbert · scripts/run_clf_finetuning.py:123

    def __init__(
            self,
            args: DataTrainingArguments,
            op_args: GeneralArguments,
            tokenizer: PreTrainedTokenizer,
            limit_length: Optional[int] = None,
            mode: Split = Split.train,
            cache_dir: Optional[str] = None,
    ):
        self.args = args
        self.processor = DataProcessor()
        self.output_mode = op_args.output_mode

        self.processor.set_labels(op_args._labels())

        if isinstance(mode, str):
            try:
                mode = Split[mode]

View source record →