"KeyError: 'text' text_column = self.column_mapping["text"]"

đŸ¤—AutoTrain
3

The issue is that Docker UI version does not work. Here the link to my spaces of AutoTrain Sheripov/autotrain-advanced which is duplicated from autotrain-projects/autotrain-advanced

image
image2794Ă—1600 352 KB

It works If locally setup it with using python version:

params = TokenClassificationParams(
    data_path="model_training/data/",
    model="obi/deid_roberta_i2b2",
    lr=3e-5,
    epochs=6, 
    max_seq_length=128,
    batch_size=32,
    warmup_ratio=0.1,
    gradient_accumulation=1,
    optimizer="adamw_torch",
    scheduler="linear",
    weight_decay=0.01,
    max_grad_norm=0.5,  # Reduce gradient norm clipping
    seed=42,
    train_split="train",
    valid_split="valid",
    tokens_column="tokens",
    tags_column="tags",
    logging_steps=5,
    project_name="deid-roberta-i2b2-fine-tuned-german",
    auto_find_batch_size=False,
    mixed_precision="none",  
    save_total_limit=2, 
    token="",
    push_to_hub=True,
    eval_strategy="epoch",
    username="",
    log="wandb",
    early_stopping_patience=3, 
    early_stopping_threshold=0.01,
)

backend = "local"
project = AutoTrainProject(params=params, backend=backend, process=True)
project.create()
1 Like