stage: 3
status: completed

dataset:
  repo_id: Hasanblanka/polygraf-ner-stage_2-dataset
  revision: bce1f32778a333297ba1bf9749c13dd394e6256d
  train_split: train
  test_split: test
  expected_train_records: 506
  expected_test_records: 127

model:
  repo_id: distilbert/distilbert-base-cased
  revision: 6ea81172465e8b0ad3fddeed32b986cdcdcffcf0
  architecture: DistilBertForTokenClassification
  parameter_count_base: 65800000
  hidden_size: 768
  hidden_layers: 6
  attention_heads: 12
  license: apache-2.0

reproducibility:
  seed: 42
  validation_split: false
  local_environment: machinelearning
  device: cpu
  deterministic_algorithms: true

data_preparation:
  max_length: 512
  truncation: false
  dynamic_padding: true
  special_token_label_id: -100
  prepared_dir: data/interim/stage3_bio
  summary_path: artifacts/stage3/bio_preparation_summary.json

training:
  epochs: 4
  train_batch_size: 8
  eval_batch_size: 8
  gradient_accumulation_steps: 2
  effective_batch_size: 16
  learning_rate: 0.00005
  weight_decay: 0.01
  warmup_ratio: 0.1
  max_grad_norm: 1.0
  logging_steps: 10
  cpu_threads: 8
  num_workers: 0

benchmark:
  batch_size: 1
  warmup_batches: 5
  measured_batches: 100

outputs:
  model_dir: models/stage3_first_model
  artifact_dir: artifacts/stage3
  train_predictions: artifacts/stage3/train_predictions.jsonl
  test_predictions: artifacts/stage3/test_predictions.jsonl
  training_summary: artifacts/stage3/training_summary.json
  training_history: artifacts/stage3/training_history.json
  train_metrics: artifacts/stage3/train_metrics.json
  test_metrics: artifacts/stage3/test_metrics.json
  inference_benchmark: artifacts/stage3/inference_benchmark.json

labels:
  entity_order:
    - PERSON
    - ORGANIZATION
    - LOCATION
    - TIMEDATE
    - PRODUCT
    - WORKOFART
    - JOB
    - AMOUNT
  bio_labels:
    - O
    - B-PERSON
    - I-PERSON
    - B-ORGANIZATION
    - I-ORGANIZATION
    - B-LOCATION
    - I-LOCATION
    - B-TIMEDATE
    - I-TIMEDATE
    - B-PRODUCT
    - I-PRODUCT
    - B-WORKOFART
    - I-WORKOFART
    - B-JOB
    - I-JOB
    - B-AMOUNT
    - I-AMOUNT