zoo.automl.model package

Subpackages

• zoo.automl.model.tcmf package
  • Submodules
  • zoo.automl.model.tcmf.DeepGLO module
  • zoo.automl.model.tcmf.data_loader module
  • zoo.automl.model.tcmf.local_model module
  • zoo.automl.model.tcmf.local_model_distributed_trainer module
  • zoo.automl.model.tcmf.time module
  • Module contents

Submodules

zoo.automl.model.MTNet_keras module

zoo.automl.model.Seq2Seq module

zoo.automl.model.Seq2Seq_pytorch module

zoo.automl.model.VanillaLSTM module

zoo.automl.model.VanillaLSTM_pytorch module

zoo.automl.model.XGBoostRegressor module

zoo.automl.model.abstract module

class zoo.automl.model.abstract.BaseModel

Bases: abc.ABC

base model for automl tuning

check_optional_config = False

evaluate(x, y, metric=None)

Evaluate the model :param x: input :param y: target :param metric: :return: a list of metric evaluation results

fit_eval(x, y, validation_data=None, mc=False, verbose=0, **config)

optimize and evaluate for one iteration for tuning :param config: tunable parameters for optimization :return:

future_seq_len = None

predict(x, mc=False)

Prediction. :param x: input :return: result

restore(model_path, **config)

restore model from model file and config. :param model_path: the model file :param config: the config :return: the restored model

save(model_path, config_path, **config)

save model to file. :param model_path: the model file path to be saved to. :param config_path: the config file path to be saved to. :return:

zoo.automl.model.tcmf_model module

zoo.automl.model.time_sequence module

Module contents