ner:
  labels:
  - 'O'
  - 'I-null'
  - 'I-BANK'
  - 'I-INDEX'
  - 'I-INVESTMENT'
  - 'B-PAYMENTCARD'
  - 'I-PAYMENTMETHOD'
  - 'B-INDEX'
  - 'I-CURRENCY'
  - 'B-null'
  - 'B-INVESTMENT'
  - 'I-BANKACCOUNT'
  - 'I-PAYMENTCARD'
  - 'B-BANKACCOUNT'
  - 'B-BANK'
  - 'B-CURRENCY'
  - 'B-PAYMENTMETHOD'
dataset:
  training: ../ner-training-data/training
  validation: ../ner-training-data/validation
transformer:
  path: /workdir/persistent/roberta-base
model:
  batch_size: 8
  initial_learn_rate: 0.0002
  epochs: 1
  stop_early: true
services:
  wandb:
    enabled: true
    config:
      project: kg_roberta_logging
      notes: "Argo Workflow"
      tags: ["argo","ner"]
artifacts:
  pretained_model_output:
    enabled: true
    path: /tmp/banking-roberta
  tensorboard_output:
    enabled: true
    path: /tmp/tf-output