relik/reader/conf/data/re.yaml

train_dataset_path: "relik/reader/data/nyt-alby+/train.jsonl"
val_dataset_path: "relik/reader/data/nyt-alby+/valid.jsonl"
test_dataset_path: "relik/reader/data/nyt-alby+/test.jsonl"

relations_definitions:  
  /people/person/nationality: "nationality"
  /sports/sports_team/location: "sports team location"
  /location/country/administrative_divisions: "administrative divisions"  
  /business/company/major_shareholders: "shareholders"
  /people/ethnicity/people: "ethnicity"
  /people/ethnicity/geographic_distribution: "geographic distributi6on"
  /business/company_shareholder/major_shareholder_of: "major shareholder" 
  /location/location/contains: "location"
  /business/company/founders: "founders"
  /business/person/company: "company"
  /business/company/advisors: "advisor"
  /people/deceased_person/place_of_death: "place of death"
  /business/company/industry: "industry"
  /people/person/ethnicity: "ethnic background"
  /people/person/place_of_birth: "place of birth"
  /location/administrative_division/country: "country of an administration division"
  /people/person/place_lived: "place lived"
  /sports/sports_team_location/teams: "sports team"
  /people/person/children: "child"
  /people/person/religion: "religion"
  /location/neighborhood/neighborhood_of: "neighborhood"
  /location/country/capital: "capital"
  /business/company/place_founded: "company founded location"
  /people/person/profession: "occupation"

train_dataset:
  _target_: "relik.reader.relik_reader_re_data.RelikREDataset"
  transformer_model: "${model.model.transformer_model}"
  materialize_samples: False
  shuffle_candidates: False
  flip_candidates: 1.0
  noise_param: 0.0
  for_inference: False
  tokens_per_batch: 4096
  min_length: -1
  special_symbols: null
  relations_definitions: ${data.relations_definitions}
  sorting_fields: 
    - "predictable_candidates"
val_dataset:
  _target_: "relik.reader.relik_reader_re_data.RelikREDataset"
  transformer_model: "${model.model.transformer_model}"
  materialize_samples: False
  shuffle_candidates: False
  flip_candidates: False
  for_inference: True
  min_length: -1
  special_symbols: null
  relations_definitions: ${data.relations_definitions}