예제 #1
0
 def train(self, *args, **kwargs):
     import comet_ml
     self.experiment = comet_ml.Experiment(log_code=False)
     self.experiment.set_code(" ".join(args))
     self.experiment.set_filename("Ludwig CLI")
     self._log_html(" ".join(args))
     config = comet_ml.get_config()
     self._save_config(config)
예제 #2
0
파일: comet.py 프로젝트: rghosh8/ludwig
    def train(self, *args, **kwargs):
        import comet_ml
        try:
            self.experiment = comet_ml.Experiment(log_code=False)
        except Exception:
            logging.error("comet_ml.Experiment() had errors. Perhaps you need to define COMET_API_KEY")
            return

        logging.info("comet.train() called......")
        cli = self._make_command_line(args)
        self.experiment.set_code(cli)
        self.experiment.set_filename("Ludwig CLI")
        self._log_html(cli)
        config = comet_ml.get_config()
        self._save_config(config)
예제 #3
0
    def train_init(self, experiment_directory, experiment_name, model_name,
                   resume, output_directory):
        if self.cometml_experiment:
            # Comet ML already initialized
            return

        import comet_ml
        try:
            self.cometml_experiment = comet_ml.Experiment(log_code=False,
                                                          project_name=experiment_name)
        except Exception:
            self.cometml_experiment = None
            logger.exception(
                "comet_ml.Experiment() had errors. Perhaps you need to define COMET_API_KEY")
            return

        logger.info("comet.train_init() called......")
        self.cometml_experiment.set_name(model_name)
        self.cometml_experiment.set_filename("Ludwig API")
        config = comet_ml.get_config()
        self._save_config(config, directory=experiment_directory)
예제 #4
0
 def train(self, *args, **kwargs):
     import comet_ml
     self.experiment = comet_ml.Experiment(log_code=False)
     config = comet_ml.get_config()
     self._save_config(config)