def train_model()

in ludwig/contribs/comet.py [0:0]

• 15 lines of code
• 6 McCabe index (conditional complexity)

    def train_model(self, model, model_definition, model_definition_path,
                    *args, **kwargs):
        logger.info("comet.train_model() called......")
        if self.cometml_experiment:
            # TODO tf2: currently no clear way to set model graph
            # see: https://github.com/comet-ml/issue-tracking/issues/296
            # if model:
            #     self.cometml_experiment.set_model_graph(
            #         str(model._graph.as_graph_def()))

            if model_definition:
                if model_definition_path:
                    base_name = os.path.basename(model_definition_path)
                else:
                    base_name = "model_definition.yaml"
                if "." in base_name:
                    base_name = base_name.rsplit(".", 1)[0] + ".json"
                else:
                    base_name = base_name + ".json"
                self.cometml_experiment.log_asset_data(model_definition,
                                                       base_name)