Advertisement
Guest User

Untitled

a guest
Nov 14th, 2019
87
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
JSON 1.14 KB | None | 0 0
  1. function (
  2.     dataset = 'semeval',
  3.     masked = true,
  4.     use_pad_text = true,
  5.     use_input_mask = true,
  6.     use_sentence_copy = true,
  7.     model = 'large',
  8.     add_bla_bla = true,
  9.     lemmatizer_strategy = 'max',
  10.     selector_k = 10
  11. )
  12. {
  13.     dataset: dataset,
  14.     configs: [{
  15.         prob_estimator: {
  16.             "module": "subst_models.xlnet_subst",
  17.             "class": "XLNetSubstEstimator",
  18.             "parameters": {
  19.                 "masked": masked,
  20.                 "use_pad_text": use_pad_text,
  21.                 "use_input_mask": use_input_mask,
  22.                 "use_sentence_copy": use_sentence_copy,
  23.                 "batch_size": 100,
  24.                 "model_name": "xlnet-" + model + "-cased",
  25.                 "add_bla_bla": add_bla_bla
  26.             }
  27.         },
  28.         lemmatizer: {
  29.             "module": "subst_models.lemmatize",
  30.             "class": "Lemmatizer",
  31.             "parameters": {
  32.                 "flag": "spacy",
  33.                 "strategy": lemmatizer_strategy
  34.             }
  35.         },
  36.         excluder: {
  37.             "module": "subst_models.target_exclude",
  38.             "class": "Excluder",
  39.             "parameters": {
  40.                 "flag": "spacy"
  41.             }
  42.         },
  43.         selector: {
  44.             "module": "evaluator.models.selectors",
  45.             "class": "TopKSelector",
  46.             "parameters": {
  47.                 "k": selector_k
  48.             }
  49.         }
  50.     }]
  51. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement