autogluon
Fast and Accurate ML in 3 Lines of Code
File Explorer
Download Latest Version (.zip)- action.yml
- action.yml
- action.yml
- action.yml
- bug_report.md
- config.yml
- doc_improvement.yml
- feature_request.md
- build_all_docs.sh
- build_cloud_fit_deploy_tutorial.sh
- build_doc.sh
- build_multimodal_tutorial.sh
- build_tabular_prediction_tutorial.sh
- build_timeseries_tutorial.sh
- copy_docs.sh
- env_setup.sh
- lint_check.sh
- run_benchmark.sh
- setup_hf_model_mirror.py
- setup_mmcv.sh
- test_common.sh
- test_core.sh
- test_features.sh
- test_install.sh
- test_install_windows.sh
- test_multimodal.sh
- test_multimodal_others.sh
- test_multimodal_others_2.sh
- test_multimodal_predictor.sh
- test_tabular.sh
- test_timeseries.sh
- version_diff.sh
- write_to_s3.sh
- benchmark-command.yml
- benchmark_master.yml
- build_latest_image.yml
- check_hf_model_list.yml
- codeguru-reviewer.yml
- codeql.yml
- codespell.yml
- continuous_integration.yml
- continuous_integration_multigpu.yaml
- lockfile_check.yml
- platform_tests-command.yml
- pypi_release.yml
- pythonpublish.yml
- pythonpublish_testpypi.yml
- README.md
- slash_command_dispatch.yml
- update-pre-commit.yml
- unittest_env.yml
- PULL_REQUEST_TEMPLATE.md
- __init__.py
- pyproject.toml
- setup.py
- autogluon_job.sh
- docker_deploy.sh
- Dockerfile.cpu
- Dockerfile.gpu
- Dockerfile.pyodide
- README.md
- cancel-job.py
- submit-job.py
- multimodal_constraints.yaml
- multimodal_frameworks_template.yaml
- tabular_full.yaml
- tabular_small.yaml
- tabular_test.yaml
- config.yaml
- constraints.yaml
- frameworks_template.yaml
- setup_hf_cache.py
- timeseries_small.yaml
- config.yaml
- constraints.yaml
- frameworks_template.yaml
- evaluate.py
- generate_amlb_user_dir.sh
- generate_bench_config.sh
- generate_framework.py
- Dockerfile.cpu-inference
- Dockerfile.cpu-training
- Dockerfile.gpu-inference
- Dockerfile.gpu-training
- full_install_image.sh
- login_ecr.sh
- deploy.sh
- Dockerfile
- download_hf_models.py
- download_hf_models.sh
- __init__.py
- feature_metadata.py
- infer_types.py
- types.py
- __init__.py
- _utils.py
- load_archive.py
- load_json.py
- load_pd.py
- load_pkl.py
- load_s3.py
- load_str.py
- load_zip.py
- __init__.py
- _model_filter.py
- __init__.py
- save_json.py
- save_pd.py
- save_pkl.py
- save_str.py
- __init__.py
- cache_presets_to_yaml.py
- compression_utils.py
- context.py
- cpu_utils.py
- cv_splitter.py
- decorators.py
- deprecated_utils.py
- distribute_utils.py
- file_utils.py
- hyperparameter_utils.py
- log_utils.py
- multiprocessing_utils.py
- nvutil.py
- pandas_utils.py
- path_converter.py
- presets_io.py
- random.py
- resource_utils.py
- s3_utils.py
- simulation_utils.py
- system_info.py
- try_import.py
- utils.py
- validation_structure.py
- warning_filter.py
- __init__.py
- constants.py
- dataset.py
- space.py
- __init__.py
- test_compare_autogluon_metadata.py
- test_cpu_detection.py
- test_cv_splitter.py
- test_get_package_versions.py
- test_pandas_utils.py
- test_presets_yaml_loading.py
- __init__.py
- test_compression_utils.py
- test_dataset.py
- test_deprecated_utils.py
- test_hash_pandas_df.py
- test_import_version.py
- test_infer_types.py
- test_load_pkl.py
- test_log_utils.py
- test_memory_limit.py
- test_model_filter.py
- test_path_converter.py
- test_s3_utils.py
- test_safe_extractall.py
- test_setup_outputdir.py
- test_version.py
- __init__.py
- conftest.py
- test_check_style.py
- pyproject.toml
- setup.py
- __init__.py
- distill_utils.py
- __init__.py
- _decision_threshold.py
- conformity_score.py
- temperature_scaling.py
- __init__.py
- _abstract_callback.py
- _early_stopping_callback.py
- _early_stopping_count_callback.py
- _early_stopping_ensemble_callback.py
- _example_callback.py
- _smooth_count.py
- __init__.py
- cleaner.py
- label_cleaner.py
- __init__.py
- constants.py
- exceptions.py
- executors.py
- ray_hpo.py
- ray_tune_constants.py
- ray_tune_scheduler.py
- ray_tune_scheduler_factory.py
- ray_tune_searcher_factory.py
- space_converter.py
- __init__.py
- abstract_learner.py
- plot_curves.py
- __init__.py
- classification_metrics.py
- quantile_metrics.py
- score_func.py
- softclass_metrics.py
- __init__.py
- _auxiliary_params.py
- _mutation_deprecated_dict.py
- _tags.py
- abstract_model.py
- abstract_nn_model.py
- model_trial.py
- __init__.py
- _dummy_quantile_regressor.py
- dummy_model.py
- __init__.py
- bagged_ensemble_model.py
- fold_fitting_strategy.py
- ray_parallel_fold_fitting_strategy.py
- stacker_ensemble_model.py
- weighted_ensemble_model.py
- __init__.py
- ensemble_selection.py
- greedy_weighted_ensemble_model.py
- __init__.py
- _utils.py
- __init__.py
- pseudolabeling.py
- __init__.py
- distributed_jobs_managers.py
- resources_calculator.py
- __init__.py
- reporter.py
- scheduler_factory.py
- seq_scheduler.py
- __init__.py
- dummy_searcher.py
- exceptions.py
- local_grid_searcher.py
- local_random_searcher.py
- local_searcher.py
- searcher_factory.py
- utils.py
- __init__.py
- global_context_snapshot.py
- __init__.py
- abstract_trainer.py
- utils.py
- __init__.py
- __init__.py
- __init__.py
- early_stopping.py
- exceptions.py
- feature_selection.py
- files.py
- infer_utils.py
- plots.py
- time.py
- utils.py
- version_utils.py
- __init__.py
- _setup_utils.py
- constants.py
- problem_type.py
- __init__.py
- test_decision_threshold.py
- __init__.py
- test_scheduler_factory.py
- __init__.py
- __init__.py
- test_ray_hpo.py
- test_space_converter.py
- __init__.py
- test_classification_metrics.py
- test_metric_kwargs.py
- test_metrics.py
- test_quantile_metrics.py
- __init__.py
- test_init_user_params.py
- test_params_aux_mutation.py
- test_params_mutation.py
- test_update_hyperparameters_with_params_trained.py
- __init__.py
- test_bagged_ensemble_model.py
- test_bagged_fit_constraints.py
- test_parallel_fold_vram.py
- __init__.py
- test_parallel_gpu_prototype.py
- test_resource_calculator.py
- __init__.py
- test_scheduler.py
- test_seq_scheduler.py
- __init__.py
- test_local_grid_searcher.py
- test_local_random_searcher.py
- test_local_searcher.py
- __init__.py
- test_presets.py
- __init__.py
- test_augment_rare_classes.py
- test_time.py
- test_utils.py
- test_version_utils.py
- __init__.py
- test_feature_selection.py
- test_import_version.py
- test_parallel_local_folding.py
- test_search_space.py
- __init__.py
- conftest.py
- test_check_style.py
- pyproject.toml
- setup.py
- autogluon-logo.jpg
- autogluon-s.png
- autogluon-w.png
- autogluon.png
- custom.css
- custom.js
- favicon.ico
- base.rst
- class.rst
- module.rst
- custom_class.rst
- autogluon.common.space.rst
- autogluon.features.rst
- autogluon.tabular.models.rst
- multimodal.rst
- tabular.rst
- timeseries.rst
- utilities.rst
- sagemaker_autopilot.png
- managed_solutions.html
- index.md
- continuous_training.ipynb
- customization.ipynb
- efficient_finetuning_basic.ipynb
- few_shot_learning.ipynb
- focal_loss.ipynb
- hyperparameter_optimization.ipynb
- index.md
- model_distillation.ipynb
- multiple_label_columns.ipynb
- presets.ipynb
- problem_types_and_metrics.ipynb
- tensorrt.ipynb
- document_classification.ipynb
- index.md
- pdf_classification.ipynb
- beginner_image_cls.ipynb
- clip_zeroshot.ipynb
- index.md
- beginner_semantic_seg.ipynb
- index.md
- beginner_multimodal.ipynb
- index.md
- multimodal-quick-start.ipynb
- multimodal_ner.ipynb
- multimodal_text_tabular.ipynb
- finetune_coco.ipynb
- index.md
- convert_data_to_coco_format.ipynb
- index.md
- object_detection_with_dataframe.ipynb
- prepare_coco17.ipynb
- prepare_pothole.ipynb
- prepare_voc.ipynb
- prepare_watercolor.ipynb
- voc_to_coco.ipynb
- index.md
- quick_start_coco.ipynb
- index.md
- image2image_matching.ipynb
- image_text_matching.ipynb
- index.md
- text2text_matching.ipynb
- text_semantic_search.ipynb
- zero_shot_img_txt_matching.ipynb
- beginner_text.ipynb
- chinese_ner.ipynb
- index.md
- multilingual_text.ipynb
- ner.ipynb
- index.md
- multimodal-faq.md
- index.md
- tabular-custom-metric.ipynb
- tabular-custom-model-advanced.ipynb
- tabular-custom-model.ipynb
- tabular-deployment.ipynb
- tabular-gpu.ipynb
- tabular-hpo.ipynb
- tabular-kaggle.ipynb
- tabular-multilabel.ipynb
- how-it-works.md
- index.md
- tabular-essentials.ipynb
- tabular-faq.md
- tabular-feature-engineering.ipynb
- tabular-foundational-models.ipynb
- tabular-indepth.ipynb
- tabular-multimodal.ipynb
- tabular-quick-start.ipynb
- forecasting-custom-model.ipynb
- index.md
- index.md
- forecasting-chronos.ipynb
- forecasting-ensembles.md
- forecasting-faq.md
- forecasting-indepth.ipynb
- forecasting-metrics.md
- forecasting-model-zoo.md
- forecasting-quick-start.ipynb
- index.md
- index.md
- v0.4.0.md
- v0.4.1.md
- v0.4.2.md
- v0.4.3.md
- v0.5.1.md
- v0.5.2.md
- v0.6.0.md
- v0.6.1.md
- v0.6.2.md
- v0.7.0.md
- v0.8.0.md
- v0.8.1.md
- v0.8.2.md
- v0.8.3.md
- v1.0.0.md
- v1.1.0.md
- v1.1.1.md
- v1.2.0.md
- v1.3.0.md
- v1.3.1.md
- v1.4.0.md
- v1.5.0.md
- v1.6.0.md
- .gitignore
- build.yml
- build_doc.sh
- build_pip_install.sh
- cheatsheet.md
- conf.py
- index.md
- install-conda-full.md
- install-cpu-pip.md
- install-cpu-source.md
- install-cpu-uv.md
- install-from-source.md
- install-gpu-pip.md
- install-gpu-source.md
- install-gpu-uv.md
- install-linux-conda-gpu.md
- install-mac-conda.md
- install-mac-cpu-source.md
- install-mac-cpu.md
- install-mac-libomp.md
- install-mac-nogpu.md
- install-modules.md
- install-windows-conda-gpu.md
- install-windows-cpu.md
- install-windows-generic.md
- install-windows-gpu.md
- install.md
- LICENSE
- README.md
- requirements_doc.txt
- versions.rst
- prepare_semantic_segmentation_datasets.py
- README.md
- run_semantic_segmentation.py
- automm_distillation_glue.py
- automm_distillation_pawsx.py
- eval_pawsx.py
- README.md
- example_kaggle_house.py
- README.md
- kaggle_feedback_prize_preprocess.py
- kaggle_feedback_prize_submit.py
- kaggle_feedback_prize_train.py
- README.md
- kaggle_pawpularity_submit.py
- kaggle_pawpularity_train.py
- README.md
- image-20221230230403853.png
- image-20221230230606344.png
- image-20221230232839988.png
- image-20221230233113119.png
- image-20221230233443558.png
- LabelStudio_export_file_reader.md
- memory_bank.py
- README.md
- utils.py
- benchmarking.py
- detection_eval.py
- detection_inference.py
- detection_load.py
- detection_train.py
- download_coco17.sh
- download_voc07.sh
- download_voc0712.sh
- download_voc12.sh
- download_watercolor.sh
- eval_pretrained_coco_format.py
- eval_pretrained_voc_format.py
- finetune_coco_format.py
- finetune_on_pothole_dataset.py
- inference_pretrained_coco_format.py
- inference_pretrained_voc_format.py
- load_predictor.py
- pretrain_objects365.py
- quick_start_on_a_tiny_dataset.py
- visualize_results.py
- feature_extraction_example.py
- onnx_text.py
- dataset.py
- example_tabular.py
- README.md
- run_all.sh
- example_cancer_survival.py
- README.md
- glue.txt
- superglue.txt
- generate_submission.py
- prepare_data_scientist_salary.sh
- prepare_glue.py
- prepare_mercari_kaggle.sh
- prepare_price_of_books.sh
- README.md
- run_competition.py
- run_glue.sh
- run_text_prediction.py
- demo.py
- example_distill_binary.py
- SHAP with AutoGluon-Tabular Census income classification.ipynb
- SHAP with AutoGluon-Tabular Diabetes regression.ipynb
- SHAP with AutoGluon-Tabular.ipynb
- example_advanced_tabular.py
- example_custom_feature_generator.py
- example_custom_model_tabular.py
- example_mitra.py
- example_quantile_regression.py
- example_simple_tabular.py
- __init__.py
- _numba_opt.py
- canonical_key.py
- combinations.py
- combinations_lite.py
- filtering.py
- memory.py
- operation.py
- preprocessor.py
- __init__.py
- _sklearn_compat.py
- _squashing_scaler.py
- __init__.py
- abstract.py
- astype.py
- auto_ml_pipeline.py
- binned.py
- bulk.py
- cat_as_num.py
- cat_int.py
- category.py
- datetime.py
- drop_duplicates.py
- drop_unique.py
- dummy.py
- fillna.py
- frequency.py
- groupby.py
- identity.py
- isnan.py
- label_encoder.py
- memory_minimize.py
- one_hot_encoder.py
- oof_target_encoder.py
- pipeline.py
- rename.py
- rsfc.py
- selection.py
- text_ngram.py
- text_special.py
- __init__.py
- _ag_feature_generator_registry.py
- _feature_generator_registry.py
- parse_custom_generator.py
- __init__.py
- binning.py
- utils.py
- vectorizers.py
- __init__.py
- test_operation.py
- __init__.py
- test_astype.py
- test_auto_ml_pipeline.py
- test_bulk.py
- test_category.py
- test_datetime.py
- test_drop_duplicates.py
- test_dummy.py
- test_fillna.py
- test_groupby.py
- test_identity.py
- test_isnan.py
- test_label_encoder.py
- test_one_hot_encoder.py
- test_oof_target_encoder.py
- test_pipeline.py
- test_rename.py
- test_selection.py
- test_text_ngram.py
- test_text_special.py
- __init__.py
- conftest.py
- test_feature_metadata.py
- test_utils.py
- __init__.py
- conftest.py
- test_check_style.py
- pyproject.toml
- setup.py
- __init__.py
- prepare_detection_dataset.py
- voc2coco.py
- default.yaml
- default.yaml
- default.yaml
- default.yaml
- default.yaml
- default.yaml
- dino-4scale_r50_8xb2-12e_coco.py
- dino-5scale_swin-l_8xb2-12e_coco.py
- dino-5scale_swin-l_8xb2-36e_coco.py
- dino_swinl_tta.py
- dino_tta.py
- __init__.py
- faster_rcnn_r50_fpn.py
- __init__.py
- faster_rcnn_r50_fpn_1x_voc0712.py
- voc0712.py
- __init__.py
- yolox_l_8xb8-300e_coco.py
- yolox_m_8xb8-300e_coco.py
- yolox_nano_8xb8-300e_coco.py
- yolox_s_8xb8-300e_coco.py
- yolox_tiny_8xb8-300e_coco.py
- yolox_tta.py
- yolox_x_8xb8-300e_coco.py
- __init__.py
- coco_detection.py
- default_runtime.py
- schedule_1x.py
- __init__.py
- __init__.py
- __init__.py
- multi_image_mix_dataset.py
- __init__.py
- process_mmdet.py
- process_mmlab_base.py
- process_mmocr.py
- __init__.py
- collator.py
- datamodule.py
- dataset.py
- infer_types.py
- label_encoder.py
- mixup.py
- nlpaug.py
- preprocess_dataframe.py
- process_categorical.py
- process_document.py
- process_image.py
- process_label.py
- process_ner.py
- process_numerical.py
- process_semantic_seg_img.py
- process_text.py
- randaug.py
- template_engine.py
- templates.py
- trivial_augmenter.py
- utils.py
- __init__.py
- base.py
- ensemble.py
- few_shot_svm.py
- matching.py
- ner.py
- object_detection.py
- semantic_segmentation.py
- modeling_sam_for_conv_lora.py
- __init__.py
- base.py
- fusion_mlp.py
- fusion_ner.py
- fusion_transformer.py
- __init__.py
- adaptation_layers.py
- augmenter.py
- categorical_mlp.py
- clip.py
- custom_transformer.py
- document_transformer.py
- ft_transformer.py
- hf_text.py
- meta_transformer.py
- mlp.py
- mmdet_image.py
- mmocr_text_detection.py
- mmocr_text_recognition.py
- ner_text.py
- numerical_mlp.py
- sam.py
- t_few.py
- timm_image.py
- utils.py
- __init__.py
- bce_loss.py
- focal_loss.py
- lemda_loss.py
- rkd_loss.py
- softmax_losses.py
- structure_loss.py
- utils.py
- __init__.py
- lr_schedulers.py
- utils.py
- __init__.py
- coverage_metrics.py
- hit_rate_metrics.py
- ranking_metrics.py
- semantic_seg_metrics.py
- utils.py
- __init__.py
- deepspeed.py
- lit_distiller.py
- lit_matcher.py
- lit_mmdet.py
- lit_module.py
- lit_ner.py
- lit_semantic_seg.py
- utils.py
- __init__.py
- cache.py
- checkpoint.py
- colormap.py
- config.py
- device.py
- distillation.py
- download.py
- env.py
- export.py
- hpo.py
- inference.py
- install.py
- label_studio.py
- load.py
- log.py
- matcher.py
- misc.py
- mmcv.py
- object_detection.py
- onnx.py
- path.py
- precision.py
- presets.py
- problem_types.py
- registry.py
- save.py
- strategy.py
- visualizer.py
- __init__.py
- constants.py
- predictor.py
- __init__.py
- test_backbones.py
- test_data_collators.py
- test_deployment_onnx.py
- test_dump_model.py
- test_ensemble.py
- test_extract_features.py
- test_load.py
- test_losses.py
- test_matcher.py
- test_metrics.py
- test_ner.py
- test_object_detection.py
- test_preprocess_data.py
- test_presets.py
- test_process_images.py
- test_process_multimodal.py
- test_process_text.py
- test_registry.py
- test_save.py
- test_semantic_segmentation.py
- __init__.py
- test_classify_doc.py
- test_config.py
- test_custom_hparams.py
- test_custom_training.py
- test_distiller.py
- test_few_shot.py
- test_hpo.py
- test_image_formats.py
- test_ner_chinese.py
- test_problem_types.py
- test_text_detection.py
- test_text_recognition.py
- test_tokenizers.py
- test_zero_shot.py
- __init__.py
- test_predictor.py
- __init__.py
- unittest_datasets.py
- utils.py
- __init__.py
- __init__.py
- conftest.py
- hf_dataset_list.yaml
- hf_model_list.yaml
- README.md
- test_check_style.py
- pyproject.toml
- setup.py
- SKILL.md
- add_links_to_release_notes.py
- ReleaseInstructions.md
- update-conda-recipes.md
- __init__.py
- zeroshot_portfolio_2023.py
- zeroshot_portfolio_2025.py
- zeroshot_portfolio_commercial_2026_08_05.py
- zeroshot_portfolio_cpu_2025_12_18.py
- zeroshot_portfolio_gpu_2025_12_18.py
- zeroshot_portfolio_noncommercial_2026_08_05.py
- __init__.py
- config_helper.py
- feature_generator_presets.py
- hyperparameter_configs.py
- pipeline_presets.py
- presets_configs.py
- __init__.py
- _scikit_mixin.py
- _tabular_classifier.py
- _tabular_regressor.py
- plot_leaderboard.py
- __init__.py
- abstract_learner.py
- default_learner.py
- __init__.py
- rapids_utils.py
- torch_utils.py
- __init__.py
- abstract_torch_model.py
- __init__.py
- automm_model.py
- ft_transformer.py
- __init__.py
- parameters.py
- searchspaces.py
- __init__.py
- callbacks.py
- catboost_model.py
- catboost_softclass_utils.py
- catboost_utils.py
- __init__.py
- parameters.py
- searchspaces.py
- __init__.py
- ebm_model.py
- __init__.py
- parameters.py
- searchspaces.py
- __init__.py
- callbacks.py
- fastai_helpers.py
- imports_helper.py
- quantile_helpers.py
- tabular_nn_fastai.py
- __init__.py
- image_predictor.py
- __init__.py
- imodels_models.py
- __init__.py
- _knn_loo_variants.py
- knn_model.py
- knn_rapids_model.py
- knn_utils.py
- __init__.py
- parameters.py
- searchspaces.py
- __init__.py
- callbacks.py
- lgb_model.py
- lgb_utils.py
- __init__.py
- parameters.py
- searchspaces.py
- __init__.py
- lr_model.py
- lr_preprocessing_utils.py
- lr_rapids_model.py
- __init__.py
- config_pretrain.py
- config_run.py
- enums.py
- tab2d.yaml
- __init__.py
- callbacks.py
- get_loss.py
- get_optimizer.py
- get_scheduler.py
- prediction_metrics.py
- trainer_finetune.py
- __init__.py
- collator.py
- dataset_finetune.py
- dataset_split.py
- preprocessor.py
- __init__.py
- base.py
- embedding.py
- tab2d.py
- __init__.py
- set_seed.py
- __init__.py
- __init__.py
- mitra_model.py
- sklearn_interface.py
- __init__.py
- nori_model.py
- __init__.py
- realmlp_model.py
- __init__.py
- native.py
- onnx.py
- __init__.py
- rf_model.py
- rf_quantile.py
- rf_rapids_model.py
- __init__.py
- tabdpt_model.py
- tabdpt_turbo_model.py
- __init__.py
- tabicl_model.py
- __init__.py
- _tabm_internal.py
- rtdl_num_embeddings.py
- tabm_model.py
- tabm_reference.py
- __init__.py
- config_run.py
- __init__.py
- callbacks.py
- collator.py
- dataset_split.py
- enums.py
- get_loss.py
- get_optimizer.py
- get_scheduler.py
- trainer_finetune.py
- y_transformer.py
- __init__.py
- dataset_finetune.py
- preprocessor.py
- __init__.py
- embedding.py
- foundation_transformer.py
- __init__.py
- __init__.py
- prediction_metrics.py
- __init__.py
- LICENSE
- README.md
- tabpfnmix_classifier.py
- tabpfnmix_regressor.py
- __init__.py
- tabpfnmix_model.py
- __init__.py
- tabpfn3_model.py
- tabpfnv2_5_model.py
- tabpfnv2_6_model.py
- __init__.py
- prep_lgb_model.py
- prep_mixin.py
- __init__.py
- native.py
- onnx.py
- __init__.py
- parameters.py
- searchspaces.py
- __init__.py
- tabular_nn_torch.py
- tabular_torch_dataset.py
- torch_network_modules.py
- __init__.py
- categorical_encoders.py
- data_preprocessor.py
- nn_architecture_utils.py
- __init__.py
- __init__.py
- text_prediction_v1_model.py
- __init__.py
- parameters.py
- searchspaces.py
- __init__.py
- callbacks.py
- xgboost_model.py
- xgboost_utils.py
- __init__.py
- xt_model.py
- __init__.py
- __init__.py
- interpretable_predictor.py
- predictor.py
- __init__.py
- _ag_model_registry.py
- _model_registry.py
- __init__.py
- fit_helper.py
- generate_datasets.py
- model_fit_helper.py
- __init__.py
- presets.py
- presets_distill.py
- __init__.py
- abstract_trainer.py
- auto_trainer.py
- __init__.py
- feature_pruner.py
- __init__.py
- __init__.py
- test_tabular_lite.py
- test_tabular_regression.py
- utils.py
- __init__.py
- test_calibrate.py
- __init__.py
- test_callbacks.py
- __init__.py
- test_config_helper.py
- test_feature_generator_presets.py
- test_pipline_presets.py
- test_presets.py
- test_validation_size_curves.py
- __init__.py
- test_label_cleaner.py
- test_learning_curves.py
- __init__.py
- test_dynamic_stacking.py
- __init__.py
- test_edgecases.py
- test_predict_proba_multi_seed.py
- __init__.py
- test_scikit_api.py
- __init__.py
- test_bagged_deterministic.py
- test_model_random_seed.py
- test_refit_full.py
- test_stack_feature_usage.py
- __init__.py
- test_advanced.py
- test_automm.py
- test_catboost.py
- test_dummy.py
- test_ebm.py
- test_image_predictor.py
- test_knn.py
- test_lightgbm.py
- test_lightgbm_prep.py
- test_linear.py
- test_mitra.py
- test_nori.py
- test_realmlp.py
- test_rf.py
- test_tabdpt.py
- test_tabdpt_turbo.py
- test_tabicl.py
- test_tabm.py
- test_tabpfn3.py
- test_tabpfnv2.py
- test_tabular_nn.py
- test_tabular_nn_fastai.py
- test_text_prediction_v1_model.py
- test_xgboost.py
- test_xt.py
- __init__.py
- pseudo_filter.py
- __init__.py
- test_model_registry.py
- __init__.py
- test_bagging_resource_allocation.py
- test_custom_memory_limit.py
- test_gpu_assignment.py
- test_hpo_resource_allocation.py
- test_resource_allocation_combined.py
- test_resources_mocking.py
- test_total_resource_allocation.py
- __init__.py
- test_gpu_memory_estimate.py
- test_tabular.py
- test_validation_structure.py
- __init__.py
- conftest.py
- test_check_style.py
- pyproject.toml
- setup.py
- __init__.py
- hyperparameter_presets.py
- predictor_presets.py
- __init__.py
- ts_dataframe.py
- __init__.py
- abstract.py
- point.py
- quantile.py
- utils.py
- __init__.py
- abstract_timeseries_model.py
- model_trial.py
- tunable.py
- __init__.py
- mlforecast.py
- per_step.py
- transforms.py
- utils.py
- __init__.py
- chronos2.py
- model.py
- utils.py
- __init__.py
- abstract.py
- linear_stacker.py
- per_quantile_tabular.py
- tabular.py
- __init__.py
- abstract.py
- models.py
- __init__.py
- abstract.py
- basic.py
- greedy.py
- __init__.py
- abstract.py
- ensemble_selection.py
- per_item_greedy.py
- __init__.py
- abstract.py
- dataset.py
- models.py
- __init__.py
- abstract_local_model.py
- naive.py
- npts.py
- statsforecast.py
- __init__.py
- multi_window_model.py
- __init__.py
- attention.py
- backbone.py
- distribution.py
- kvcache.py
- rope.py
- rotary_embedding_torch.py
- scaler.py
- transformer.py
- __init__.py
- dataset.py
- forecaster.py
- LICENSE
- README.txt
- __init__.py
- dataloader.py
- hf_pretrained_model.py
- model.py
- __init__.py
- backbone.py
- configuration.py
- LICENSE
- model.py
- README.txt
- unit_scaling.py
- __init__.py
- dataloader.py
- model.py
- __init__.py
- registry.py
- __init__.py
- ensemble_composer.py
- model_set_builder.py
- prediction_cache.py
- trainer.py
- utils.py
- __init__.py
- covariate_scaler.py
- target_scaler.py
- __init__.py
- base.py
- lags.py
- seasonality.py
- time_features.py
- __init__.py
- constants.py
- features.py
- forecast.py
- timer.py
- warning_filters.py
- __init__.py
- learner.py
- predictor.py
- regressor.py
- splitter.py
- __init__.py
- test_all_models.py
- __init__.py
- test_chronos2.py
- test_export.py
- test_model.py
- test_utils.py
- __init__.py
- conftest.py
- test_abstract.py
- test_linear_stacker.py
- test_models.py
- test_tabular.py
- __init__.py
- conftest.py
- test_abstract.py
- test_per_item_greedy.py
- test_weighted.py
- __init__.py
- common.py
- conftest.py
- test_abstract.py
- test_gluonts.py
- test_local.py
- test_mlforecast.py
- test_models.py
- test_multi_window_model.py
- test_per_step_tabular.py
- test_registry.py
- test_toto.py
- test_toto2.py
- __init__.py
- test_ensemble_composer.py
- test_model_set_builder.py
- test_prediction_cache.py
- test_trainer.py
- __init__.py
- test_features.py
- test_timer.py
- test_utils.py
- __init__.py
- common.py
- conftest.py
- test_learner.py
- test_metrics.py
- test_predictor.py
- test_regressor.py
- test_splitter.py
- test_transforms.py
- test_ts_dataset.py
- __init__.py
- conftest.py
- pyproject.toml
- setup.py
- .gitattributes
- .gitignore
- .pre-commit-config.yaml
- .python-version
- AWESOME.md
- CITING.md
- CODE_OF_CONDUCT.md
- CONTRIBUTING.md
- full_install.bat
- full_install.sh
- LICENSE
- NOTICE
- pyproject.toml
- README.md
- SECURITY.md
- setup.cfg
- uv.lock
- VERSION
๐ Installation Guide
1. Get the code
git clone https://github.com/autogluon/autogluon
Downloads the entire project code from GitHub to your computer.
cd autogluon
Moves into the project folder you just downloaded.
2. Official Install Script
Easy RecommendedPrerequisites
- Python 3 Python is required to use pip.
pip install autogluon
Installs the package published on PyPI directly โ no need to clone the source.
After installing, open a new terminal and run the program's version command (e.g. --version) to confirm it worked.
Pulled directly from this repo's README.
3. Docker
EasyPrerequisites
- Git Needed to download the project code from GitHub.
- Docker Desktop Needed to build and run containers. Install it and keep it running in the background.
docker build -f CI/batch/docker/Dockerfile.cpu -t autogluon .
Builds a runnable image based on the Dockerfile.
docker run -p 8080:80 autogluon
Runs the built image as an actual container.
Run docker compose ps to check the containers are Up. If the README mentions a port, open http://localhost:PORT in your browser.
4. Python
EasyPrerequisites
pip install autogluon
Installs the package published on PyPI directly โ no need to clone the source.
If it runs without errors and prints output in the terminal, it worked.
Pulled directly from this repo's README.
// repository documentation
Was this content helpful?
(0 ratings)