def get_feature_specs(self):
feature_specs = [
feature_spec.FeatureSpec(self.name + "_eid", []),
feature_spec.FeatureSpec(self.name + "_targets", [],
is_int_feature=False)
]
return feature_specs
def get_feature_specs(self):
return [
feature_spec.FeatureSpec(self.name + "_eid", []),
feature_spec.FeatureSpec(self.name + "_start_positions", []),
feature_spec.FeatureSpec(self.name + "_end_positions", []),
feature_spec.FeatureSpec(self.name + "_is_impossible", []),
]
def get_feature_specs(self):
return [
feature_spec.FeatureSpec(self.name + "_eid", []),
feature_spec.FeatureSpec(self.name + "_labels",
[self.config.max_seq_length]),
feature_spec.FeatureSpec(self.name + "_labels_mask",
[self.config.max_seq_length],
is_int_feature=False),
feature_spec.FeatureSpec(self.name + "_labeled_positions",
[self.config.max_seq_length]),
]
def get_feature_specs(self):
shape = [self.config.max_options_num * self.config.evidences_top_k * self.config.max_seq_length]
return [
feature_spec.FeatureSpec("input_ids", shape),
feature_spec.FeatureSpec("input_mask", shape),
feature_spec.FeatureSpec("segment_ids", shape),
feature_spec.FeatureSpec("task_id", []),
feature_spec.FeatureSpec(self.name + "_eid", []),
feature_spec.FeatureSpec(self.name + "_answer_mask", [2 ** self.config.max_options_num]),
feature_spec.FeatureSpec(self.name + "_answer_ids", [2 ** self.config.max_options_num]),
feature_spec.FeatureSpec(self.name + "_answer_ids_raw", [self.config.max_options_num]),
]
def get_feature_specs(self):
return [
feature_spec.FeatureSpec(self.name + "_eid", []),
feature_spec.FeatureSpec(self.name + "_start_positions", []),
feature_spec.FeatureSpec(self.name + "_end_positions", []),
feature_spec.FeatureSpec(self.name + "_is_impossible", []),
feature_spec.FeatureSpec(self.name + "_plau_answer_start", []),
feature_spec.FeatureSpec(self.name + "_plau_answer_end", []),
# feature_spec.FeatureSpec(self.name + "_dep_mask_x", [self.dep_max]),
# feature_spec.FeatureSpec(self.name + "_dep_mask_y", [self.dep_max]),
# feature_spec.FeatureSpec(self.name + "_dep_mask_len", []),
]
def get_feature_specs(self):
return [
feature_spec.FeatureSpec(self.name + "_eid", []),
feature_spec.FeatureSpec(self.name + "_label_ids", [])
]
