tabensemb.model.CatEmbed ======================== .. currentmodule:: tabensemb.model .. autoclass:: CatEmbed :show-inheritance: .. rubric:: Methods .. automethod:: __init__ .. autosummary:: ~CatEmbed.required_models ~CatEmbed._conditional_validity ~CatEmbed._get_model_names ~CatEmbed._get_program_name ~CatEmbed._initial_values ~CatEmbed._new_model ~CatEmbed._prepare_custom_datamodule ~CatEmbed._run_custom_data_module ~CatEmbed._space .. HACK -- the point here is that we don't want this to appear in the output, but the autosummary should still generate the pages. .. autosummary:: :toctree: CatEmbed._bayes_eval CatEmbed._check_params CatEmbed._check_space CatEmbed._check_train_status CatEmbed._conditional_validity CatEmbed._custom_training_params CatEmbed._data_preprocess CatEmbed._default_metric_sklearn CatEmbed._generate_dataset CatEmbed._generate_dataset_for_required_models CatEmbed._generate_dataset_from_tensors CatEmbed._get_model_names CatEmbed._get_params CatEmbed._get_program_name CatEmbed._get_required_models CatEmbed._initial_values CatEmbed._mkdir CatEmbed._new_model CatEmbed._pred_single_model CatEmbed._predict CatEmbed._predict_all CatEmbed._predict_model CatEmbed._predict_model_on_partition CatEmbed._prepare_custom_datamodule CatEmbed._prepare_tensors CatEmbed._run_custom_data_module CatEmbed._space CatEmbed._train CatEmbed._train_data_preprocess CatEmbed._train_single_model CatEmbed._update_optimizer_lr_scheduler_params CatEmbed.cal_feature_importance CatEmbed.cal_shap CatEmbed.count_params CatEmbed.detach_model CatEmbed.fit CatEmbed.get_full_name_from_required_model CatEmbed.get_model_names CatEmbed.inspect_attr CatEmbed.new_model CatEmbed.predict CatEmbed.predict_proba CatEmbed.required_models CatEmbed.reset CatEmbed.save_kwargs CatEmbed.set_path CatEmbed.train .. HACK -- the point here is that we don't want this to appear in the output, but the autosummary should still generate the pages. .. autosummary:: :toctree: CatEmbed._support_warm_start CatEmbed._trained CatEmbed.device