Python remap_spec 예제들

예제 #1

파일: t5.py 프로젝트: oceanfly/lit

 def output_spec(self):
     spec = lit_types.remap_spec(self.model.output_spec(),
                                 self.FIELD_RENAMES)
     spec["rougeL"] = lit_types.Scalar()
     return spec

예제 #2

파일: t5.py 프로젝트: oceanfly/lit

 def input_spec(self):
     spec = lit_types.remap_spec(self.model.input_spec(),
                                 self.FIELD_RENAMES)
     spec["source_language"] = lit_types.CategoryLabel()
     spec["target_language"] = lit_types.CategoryLabel()
     return spec

예제 #3

파일: t5.py 프로젝트: oceanfly/lit

 def input_spec(self):
     return lit_types.remap_spec(self.model.input_spec(),
                                 self.FIELD_RENAMES)

예제 #4

파일: t5.py 프로젝트: PAIR-code/lit

 def input_spec(self):
     return lit_types.remap_spec(self.wrapped.input_spec(),
                                 self.FIELD_RENAMES)