jmteb_config/tasks/jsts.jsonnet

{
  jsts: {
    class_path: 'STSEvaluator',
    init_args: {
      sentence1_prefix: 'クエリ: ',
      sentence2_prefix: 'クエリ: ',
      val_dataset: {
        class_path: 'HfSTSDataset',
        init_args: {
          path: 'sbintuitions/JMTEB',
          split: 'train',
          name: 'jsts',
        },
      },
      test_dataset: {
        class_path: 'HfSTSDataset',
        init_args: {
          path: 'sbintuitions/JMTEB',
          split: 'test',
          name: 'jsts',
        },
      },
    },
  },
}