ner:
  labels:
  - "O"
  - "B-BANK"
  - "I-BANK"
  - "B-STOCKMARKETINDEX"
  - "I-STOCKMARKETINDEX"
  - "B-CURRENCY"
  - "I-CURRENCY"
  - "B-CRYPTOCURRENCY"
  - "I-CRYPTOCURRENCY"
  - "B-FUND"
  - "I-FUND"
  - "B-STOCKEXCHANGE"
  - "I-STOCKEXCHANGE"

dataset:
  training: ../ner-training-data/training
  validation: ../ner-training-data/validation
transformer:
  path: /workdir/persistent/roberta-base
checkpoints:
  enabled: true
  save_only_best: true
model:
  batch_size: 8
  initial_learn_rate: 0.00005
  epochs: 8
  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: /workdir/models/banking-roberta
  tensorboard_output:
    enabled: true
    path: /tensorboard