INNER CODE UNIT · Python

_labels

hooshvare/parsbert · scripts/run_clf_finetuning.py:62

    def _labels(self):
        if self.output_mode == 'classification':
            return [str(l) for l in list(set(self.labels.split(',')))]

        return []


@dataclass
class ModelArguments:
    """Arguments pertaining to which model/config/tokenizer we are going to fine-tune from."""

    model_name_or_path: str = field(
        metadata={"help": "Path to pretrained model or model identifier from huggingface.co/models"}
    )
    config_name: Optional[str] = field(
        default=None, metadata={"help": "Pretrained config name or path if not the same as model_name"}
    )
    tokenizer_name: Optional[str] = field(

View source record →