config.yaml

paths:
  artifacts: artifacts
  data_store: data
  logs: logs
  model_weight: model_weight
  serialized_objects: serialized_objects

data_ingestion_config:
  dataset_name: xtreme
  subset_name: PAN-X.en

data_validation_config:
  data_split:
    - train
    - test
    - validation

  type_check:
    - 

  null_check:

  columns_check:
    - tokens
    - ner_tags
    - langs

Data_preprocessing:
  ner_tags:
    - O
    - B-PER
    - I-PER
    - B-ORG
    - I-ORG
    - B-LOC
    - I-LOC

Base_model_config:
  base_model_name: xlm-roberta-base
  num_classes : 7
  num_epochs: 1
  batch_size: 16
  save_steps: 100000

Predict_model_config:
  truncation: True
  is_split_into_words: True