Ejemplo n.º 1
0
 def setUp(self):
     self.maxDiff = None
     super(TestRunConfigSerializer, self).setUp()
     reset_register_and_logger()
     self.configs_to_log, self.config_serialized_must = _get_configs_to_log(
     )
     self.configs_to_log_copy = copy.copy(self.configs_to_log)
Ejemplo n.º 2
0
 def setUp(self):
     reset_register_and_logger()
     register_new_class("data_filter1", _DummyDataFilter)
     register_new_class("file_list1", FileListDummy)
     register_new_class("reader1", _DataReaderTF)
     register_new_class("reader2", _DataReaderTF)
     register_new_class("processor1", _DataProcessorTF)
     register_new_class("processor2", _DataProcessorTF)
Ejemplo n.º 3
0
    def setUp(self):
        reset_register_and_logger()
        register_new_class("plugin1", _KPIPlugin)
        register_new_class('plugin2', _KPIPlugin)
        register_new_class("accumulator1", _KPIAccumulator)
        register_new_class("accumulator2", _KPIAccumulator)

        self.plugins_and_accumulators_configs = [
            {
                "class_name": "plugin1",
                "parameter1": 10
            },
            {
                "class_name": "plugin2",
                "parameter1": 20,
                "cachers": [{
                    "class_name": "kpi_builder_test._KPICacher"
                }]
            },
            {
                "class_name": "accumulator1",
                "parameter2": 100,
                "cachers": [{
                    "class_name": "kpi_builder_test._KPICacher",
                    "p": 10
                }],
                "savers": [{
                    "class_name": "kpi_builder_test._KPISaver",
                    "p3": 5
                }]
            },
            {
                "class_name": "accumulator2",
                "parameter2": 200,
                "savers": [{
                    "class_name": "kpi_builder_test._KPISaver2",
                    "p4": 50
                }]
            },
        ]
        self.project_global_config = {
            "KPIPlugin": {
                "cachers": [{
                    "class_name": "kpi_builder_test._KPICacher2",
                    "p2": 5
                }],
                "savers": [{
                    "class_name": "kpi_builder_test._KPISaver2",
                    "p4": 100
                }]
            }
        }
        project_global_config.clear_project_global_config()
        super().setUp()
Ejemplo n.º 4
0
 def setUp(self):
     reset_register_and_logger()
     register_new_class("dataset_with_file_list", _DatasetWithFileList)
     register_new_class("dataset_without_file_list",
                        _DatasetWithoutFileList)
     register_new_class("FileListDummy",
                        FileListDummy)
     self.number_of_files1 = 20
     self.number_of_files2 = 10
     self.file_names1 = {'data': ['input_1_{:03d}.ext'.format(i)
                                  for i in range(self.number_of_files1)]}
     self.file_names2 = {'data': ['input2_1_{:03d}.ext'.format(i)
                                  for i in range(self.number_of_files2)]}
Ejemplo n.º 5
0
    def setUp(self):
        reset_register_and_logger()

        register_new_class('dummy_dataset_train',
                           nc7.data.Dataset)
        register_new_class('dummy_dataset_eval',
                           nc7.data.Dataset)
        register_new_class('dummy_callback_train',
                           nc7.coordinator.CoordinatorCallback)
        register_new_class('dummy_plugin_train',
                           nc7.model.ModelPlugin)
        register_new_class('dummy_loss',
                           nc7.model.ModelLoss)
        register_new_class('dummy_pp',
                           nc7.model.ModelPostProcessor)
        register_new_class('dummy_metrics',
                           nc7.model.ModelMetric)
        register_new_class('dummy_summaries',
                           nc7.model.ModelSummary)
        register_new_class('dummy_dataset_fl_train',
                           _DatasetFileListDummy)
        register_new_class('dummy_dataset_fl_eval',
                           _DatasetFileListDummy)
        register_new_class('dummy_file_list',
                           FileListDummy)

        register_new_class('dummy_data_feeder',
                           nc7.data.DataFeeder)
        register_new_class('dummy_callback_infer',
                           nc7.coordinator.CoordinatorCallback)
        register_new_class('dummy_data_feeder_fl',
                           _DatafeederFileListDummy)
        register_new_class('dummy_file_list_infer',
                           FileListDummy)

        register_new_class("dummy_kpi_plugin1",
                           nc7.kpi.KPIPlugin)
        register_new_class("dummy_kpi_plugin2",
                           nc7.kpi.KPIPlugin)
        register_new_class("dummy_kpi_accumulator",
                           nc7.kpi.KPIAccumulator)
Ejemplo n.º 6
0
    def setUp(self):
        reset_register_and_logger()
        self.number_of_files1 = 20
        self.number_of_files2 = 10
        self.number_of_files3 = 10
        self.file_names1 = {
            'input1': [
                'input_1_{:03d}.ext'.format(i)
                for i in range(self.number_of_files1)
            ],
            'input2': [
                'input_2_{:03d}.ext'.format(i)
                for i in range(self.number_of_files1)
            ]
        }
        self.file_names2 = {
            'input1': [
                'input2_1_{:03d}.ext'.format(i)
                for i in range(self.number_of_files2)
            ],
            'input2': [
                'input2_2_{:03d}.ext'.format(i)
                for i in range(self.number_of_files2)
            ]
        }

        self.file_names3 = {
            'input1': [
                'input3_1_{:03d}.ext'.format(i)
                for i in range(self.number_of_files3)
            ],
            'input3': [
                'input_3_{:03d}.ext'.format(i)
                for i in range(self.number_of_files3)
            ]
        }
        self.file_names_1_2_3 = self._add_file_names()
Ejemplo n.º 7
0
 def setUp(self) -> None:
     reset_register_and_logger()
     register_new_class("data_filter1", _DummyDataFilter1)
     register_new_class("data_filter2", _DummyDataFilter2)
     self.config1 = {"class_name": "data_filter1", "p1": 20, "p2": 500}
     self.config2 = {"class_name": "data_filter2", "p1": 100}
Ejemplo n.º 8
0
 def setUp(self):
     super(TestCallbackBuilder, self).setUp()
     reset_register_and_logger()
Ejemplo n.º 9
0
 def setUp(self):
     reset_register_and_logger()
Ejemplo n.º 10
0
 def setUp(self):
     tf.reset_default_graph()
     reset_register_and_logger()
     np.random.seed(5467)