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", []) ]