File size: 2,197 Bytes
2f044c1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
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}
|