config.yaml

head_type: claim
feature_extractor:
- name: luh.feature_extractors.basic_hidden_states
  layer_nums:
  - -1
  - -5
  - -10
  - -20
- name: luh.feature_extractors.token_probabilities
  top_n: 1
uncertainty_head:
  head_dim: 4096
  n_layers: 1
  n_heads: 16
  dropout: 0.1