コード例 #1
0
ファイル: component_test.py プロジェクト: vsatyav007/tfx
 def testConstructWithTunerFn(self):
     tuner = component.Tuner(examples=self.examples,
                             transform_graph=self.transform_graph,
                             train_args=self.train_args,
                             eval_args=self.eval_args,
                             tuner_fn='path.to.tuner_fn')
     self._verify_output(tuner)
コード例 #2
0
ファイル: component_test.py プロジェクト: vsatyav007/tfx
 def testConstructDuplicateUserModule(self):
     with self.assertRaises(ValueError):
         _ = component.Tuner(examples=self.examples,
                             schema=self.schema,
                             train_args=self.train_args,
                             eval_args=self.eval_args,
                             module_file='/path/to/module/file',
                             tuner_fn='path.to.tuner_fn')
コード例 #3
0
ファイル: component_test.py プロジェクト: vsatyav007/tfx
 def testConstructWithModuleFile(self):
     tuner = component.Tuner(examples=self.examples,
                             schema=self.schema,
                             train_args=self.train_args,
                             eval_args=self.eval_args,
                             tune_args=self.tune_args,
                             module_file='/path/to/module/file')
     self._verify_output(tuner)
コード例 #4
0
 def testConstructWithCustomConfig(self):
     tuner = component.Tuner(examples=self.examples,
                             transform_graph=self.transform_graph,
                             train_args=self.train_args,
                             eval_args=self.eval_args,
                             module_file='/path/to/module/file',
                             custom_config={'key': 'value'})
     self._verify_output(tuner)