Python KerasBaseModel示例

示例#1

文件： model_builder.py 项目： xiejinglei/analytics-zoo

    def build_from_ckpt(self, checkpoint_filename):
        '''Restore from a saved model'''
        if self.backend == "pytorch":
            from zoo.automl.model.base_pytorch_model import PytorchBaseModel
            model = PytorchBaseModel(**self.params)
            model.restore(checkpoint_filename)
            return model

        elif self.backend == "keras":
            from zoo.automl.model.base_keras_model import KerasBaseModel
            model = KerasBaseModel(**self.params)
            model.restore(checkpoint_filename)
            return model

示例#2

文件： model_builder.py 项目： leonardozcm/analytics-zoo

    def build(self, config):
        if self.backend == "pytorch":
            from zoo.automl.model.base_pytorch_model import PytorchBaseModel
            model = PytorchBaseModel(**self.params)
            model.build(config)
            return model

        elif self.backend == "keras":
            from zoo.automl.model.base_keras_model import KerasBaseModel
            model = KerasBaseModel(**self.params)
            model.build(config)
            return model

        elif self.cls is not None:
            return self.cls(config=config)

        else:
            builder = self.from_name(config["model"],
                                     dev_option=config["dev_option"])
            return builder.cls(config=config)

示例#3

 def build_from_ckpt(self, checkpoint_filename):
     from zoo.automl.model.base_keras_model import KerasBaseModel
     model = KerasBaseModel(self.model_creator)
     model.restore(checkpoint_filename)
     return model

示例#4

 def build(self, config):
     from zoo.automl.model.base_keras_model import KerasBaseModel
     model = KerasBaseModel(self.model_creator)
     model.build(config)
     return model