├── .gitignore ├── .gitmodules ├── README.md ├── analyses ├── add_uninformative_features.R ├── benchmark_plots_categorical.R ├── benchmark_plots_numerical.R ├── benchmark_tab_pfn.R ├── benchmark_time.R ├── datasets │ ├── categorical_classif.csv │ ├── categorical_regression.csv │ ├── numerical_classif.csv │ └── numerical_regression.csv ├── high_frequency.R ├── hyperparameters_trees_comparison.R ├── plot_bo.R ├── plot_utils.R ├── plots │ └── random_search_classif_numerical_thumbnail.jpg ├── random_rotation.R ├── random_search_utils.R ├── rebuttal.R ├── remove_uninformative_features.R ├── results │ └── README.md ├── sankey.py ├── sankey_categorical_regression_large.svg ├── shankey.R ├── tabpfn.R └── uninformative_features.R ├── data ├── .gitignore └── data_utils.py ├── requirements.txt ├── src ├── check_difficulty.py ├── check_difficulty_custom.py ├── configs │ ├── __init__.py │ ├── all_model_configs.py │ ├── model_configs │ │ ├── __init__.py │ │ ├── ft_transformer_config.py │ │ ├── gpt_config.py │ │ ├── hgbt_config.py │ │ ├── mlp_config.py │ │ ├── resnet_config.py │ │ ├── rf_config.py │ │ ├── saint_config.py │ │ ├── skorch_config.py │ │ ├── template_config.py │ │ └── xgb_config.py │ └── wandb_config.py ├── create_models.py ├── data_transforms.py ├── export_data.py ├── generate_data.py ├── generate_dataset_pipeline.py ├── launch_benchmarks │ ├── README.md │ ├── launch_agent.sh │ ├── launch_agent_gpu.sh │ ├── launch_benchmarks.py │ ├── launch_on_cluster.py │ ├── launch_xps.py │ ├── results │ │ └── file │ ├── sweeps │ │ ├── benchmark_sweeps.csv │ │ └── xps_sweeps.csv │ └── utils.py ├── models │ ├── TabSurvey │ │ ├── .gitignore │ │ ├── Dockerfile │ │ ├── LICENSE │ │ ├── README.md │ │ ├── __init__.py │ │ ├── attributions.py │ │ ├── evaluate.py │ │ ├── insights.py │ │ ├── models │ │ │ ├── __init__.py │ │ │ ├── baseline_models.py │ │ │ ├── basemodel.py │ │ │ ├── basemodel_torch.py │ │ │ ├── danet.py │ │ │ ├── danet_lib │ │ │ │ ├── DAN_Task.py │ │ │ │ ├── __init__.py │ │ │ │ ├── abstract_model.py │ │ │ │ ├── lib │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── callbacks.py │ │ │ │ │ ├── logger.py │ │ │ │ │ ├── metrics.py │ │ │ │ │ ├── multiclass_utils.py │ │ │ │ │ └── utils.py │ │ │ │ ├── main.py │ │ │ │ ├── model │ │ │ │ │ ├── AcceleratedModule.py │ │ │ │ │ ├── DANet.py │ │ │ │ │ ├── __init__.py │ │ │ │ │ └── sparsemax.py │ │ │ │ └── predict.py │ │ │ ├── deepfm.py │ │ │ ├── deepfm_lib │ │ │ │ ├── __init__.py │ │ │ │ ├── inputs.py │ │ │ │ ├── layers │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── activation.py │ │ │ │ │ ├── core.py │ │ │ │ │ ├── interaction.py │ │ │ │ │ ├── sequence.py │ │ │ │ │ └── utils.py │ │ │ │ └── models │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── basemodel.py │ │ │ │ │ └── deepfm.py │ │ │ ├── deepgbm.py │ │ │ ├── deepgbm_lib │ │ │ │ ├── __init__.py │ │ │ │ ├── config.py │ │ │ │ ├── main.py │ │ │ │ ├── models │ │ │ │ │ ├── CatNN.py │ │ │ │ │ ├── DeepGBM.py │ │ │ │ │ ├── EmbeddingModel.py │ │ │ │ │ ├── GBDT2NN.py │ │ │ │ │ └── __init__.py │ │ │ │ ├── preprocess │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── preprocessing_cat.py │ │ │ │ │ └── preprocessing_num.py │ │ │ │ ├── trainModel.py │ │ │ │ └── utils │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── gbdt.py │ │ │ │ │ ├── helper.py │ │ │ │ │ └── tree_model_interpreter.py │ │ │ ├── dnf.py │ │ │ ├── dnf_lib │ │ │ │ ├── DNFNet │ │ │ │ │ ├── DNFNetModels │ │ │ │ │ │ ├── DNFNetComponents.py │ │ │ │ │ │ ├── __init__.py │ │ │ │ │ │ ├── model1.py │ │ │ │ │ │ ├── model100.py │ │ │ │ │ │ ├── model101.py │ │ │ │ │ │ ├── model2.py │ │ │ │ │ │ ├── model3.py │ │ │ │ │ │ ├── model4.py │ │ │ │ │ │ ├── model5.py │ │ │ │ │ │ └── model6.py │ │ │ │ │ ├── ModelHandler.py │ │ │ │ │ └── __init__.py │ │ │ │ ├── Utils │ │ │ │ │ ├── NumpyGenerator.py │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── experiment_utils.py │ │ │ │ │ └── file_utils.py │ │ │ │ ├── __init__.py │ │ │ │ └── config.py │ │ │ ├── mlp.py │ │ │ ├── modeltree.py │ │ │ ├── neural_additive_models.py │ │ │ ├── node.py │ │ │ ├── node_lib │ │ │ │ ├── __init__.py │ │ │ │ ├── arch.py │ │ │ │ ├── data.py │ │ │ │ ├── nn_utils.py │ │ │ │ ├── odst.py │ │ │ │ ├── trainer.py │ │ │ │ └── utils.py │ │ │ ├── rln.py │ │ │ ├── saint.py │ │ │ ├── saint_lib │ │ │ │ ├── __init__.py │ │ │ │ ├── augmentations.py │ │ │ │ ├── data_openml.py │ │ │ │ └── models │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── model.py │ │ │ │ │ ├── pretrainmodel.py │ │ │ │ │ └── pretrainmodel_vision.py │ │ │ ├── stg_lib │ │ │ │ ├── __init__.py │ │ │ │ ├── io.py │ │ │ │ ├── layers.py │ │ │ │ ├── losses.py │ │ │ │ ├── matching.py │ │ │ │ ├── meter.py │ │ │ │ ├── models.py │ │ │ │ ├── stg.py │ │ │ │ └── utils.py │ │ │ ├── stochastic_gates.py │ │ │ ├── tabnet.py │ │ │ ├── tabtransformer.py │ │ │ ├── tree_models.py │ │ │ └── vime.py │ │ ├── testall.sh │ │ ├── train.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── baseline_attributions.py │ │ │ ├── io_utils.py │ │ │ ├── load_data.py │ │ │ ├── parser.py │ │ │ ├── scorer.py │ │ │ └── timer.py │ ├── __init__.py │ ├── skorch_models.py │ ├── skorch_models_regression.py │ └── tabular │ │ ├── .gitignore │ │ ├── LICENSE │ │ ├── README.md │ │ ├── bin │ │ ├── analysis_ft_transformer_vs_resnet.py │ │ ├── autoint.py │ │ ├── catboost_.py │ │ ├── create_synthetic_data_plots.ipynb │ │ ├── dcn2.py │ │ ├── ensemble.py │ │ ├── ft_transformer.py │ │ ├── grownet.py │ │ ├── lightgbm_.py │ │ ├── mlp.py │ │ ├── node.py │ │ ├── report.ipynb │ │ ├── resnet.py │ │ ├── snn.py │ │ ├── tabnet.py │ │ ├── tune.py │ │ └── xgboost_.py │ │ ├── lib │ │ ├── __init__.py │ │ ├── data.py │ │ ├── deep.py │ │ ├── env.py │ │ ├── metrics.py │ │ ├── node │ │ │ ├── __init__.py │ │ │ ├── arch.py │ │ │ ├── nn_utils.py │ │ │ ├── odst.py │ │ │ └── utils.py │ │ ├── synthetic_data.py │ │ └── util.py │ │ ├── pyproject.toml │ │ ├── requirements.txt │ │ └── setup.cfg ├── preprocessing │ ├── preprocessing.py │ └── utils.py ├── run_experiment.py ├── target_function_classif.py ├── train.py └── utils │ ├── __init__.py │ ├── keyword_to_function_conversion.py │ ├── skorch_utils.py │ ├── tree.py │ └── utils.py └── tests ├── __init__.py ├── test_models.py └── test_tree_quantiles.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/.gitmodules -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/README.md -------------------------------------------------------------------------------- /analyses/add_uninformative_features.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/add_uninformative_features.R -------------------------------------------------------------------------------- /analyses/benchmark_plots_categorical.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/benchmark_plots_categorical.R -------------------------------------------------------------------------------- /analyses/benchmark_plots_numerical.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/benchmark_plots_numerical.R -------------------------------------------------------------------------------- /analyses/benchmark_tab_pfn.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/benchmark_tab_pfn.R -------------------------------------------------------------------------------- /analyses/benchmark_time.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/benchmark_time.R -------------------------------------------------------------------------------- /analyses/datasets/categorical_classif.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/datasets/categorical_classif.csv -------------------------------------------------------------------------------- /analyses/datasets/categorical_regression.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/datasets/categorical_regression.csv -------------------------------------------------------------------------------- /analyses/datasets/numerical_classif.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/datasets/numerical_classif.csv -------------------------------------------------------------------------------- /analyses/datasets/numerical_regression.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/datasets/numerical_regression.csv -------------------------------------------------------------------------------- /analyses/high_frequency.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/high_frequency.R -------------------------------------------------------------------------------- /analyses/hyperparameters_trees_comparison.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/hyperparameters_trees_comparison.R -------------------------------------------------------------------------------- /analyses/plot_bo.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/plot_bo.R -------------------------------------------------------------------------------- /analyses/plot_utils.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/plot_utils.R -------------------------------------------------------------------------------- /analyses/plots/random_search_classif_numerical_thumbnail.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/plots/random_search_classif_numerical_thumbnail.jpg -------------------------------------------------------------------------------- /analyses/random_rotation.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/random_rotation.R -------------------------------------------------------------------------------- /analyses/random_search_utils.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/random_search_utils.R -------------------------------------------------------------------------------- /analyses/rebuttal.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/rebuttal.R -------------------------------------------------------------------------------- /analyses/remove_uninformative_features.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/remove_uninformative_features.R -------------------------------------------------------------------------------- /analyses/results/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/results/README.md -------------------------------------------------------------------------------- /analyses/sankey.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/sankey.py -------------------------------------------------------------------------------- /analyses/sankey_categorical_regression_large.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/sankey_categorical_regression_large.svg -------------------------------------------------------------------------------- /analyses/shankey.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/shankey.R -------------------------------------------------------------------------------- /analyses/tabpfn.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/tabpfn.R -------------------------------------------------------------------------------- /analyses/uninformative_features.R: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/analyses/uninformative_features.R -------------------------------------------------------------------------------- /data/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/data/.gitignore -------------------------------------------------------------------------------- /data/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/data/data_utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/check_difficulty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/check_difficulty.py -------------------------------------------------------------------------------- /src/check_difficulty_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/check_difficulty_custom.py -------------------------------------------------------------------------------- /src/configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/configs/all_model_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/configs/all_model_configs.py -------------------------------------------------------------------------------- /src/configs/model_configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/configs/model_configs/ft_transformer_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/configs/model_configs/ft_transformer_config.py -------------------------------------------------------------------------------- /src/configs/model_configs/gpt_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/configs/model_configs/gpt_config.py -------------------------------------------------------------------------------- /src/configs/model_configs/hgbt_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/configs/model_configs/hgbt_config.py -------------------------------------------------------------------------------- /src/configs/model_configs/mlp_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/configs/model_configs/mlp_config.py -------------------------------------------------------------------------------- /src/configs/model_configs/resnet_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/configs/model_configs/resnet_config.py -------------------------------------------------------------------------------- /src/configs/model_configs/rf_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/configs/model_configs/rf_config.py -------------------------------------------------------------------------------- /src/configs/model_configs/saint_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/configs/model_configs/saint_config.py -------------------------------------------------------------------------------- /src/configs/model_configs/skorch_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/configs/model_configs/skorch_config.py -------------------------------------------------------------------------------- /src/configs/model_configs/template_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/configs/model_configs/template_config.py -------------------------------------------------------------------------------- /src/configs/model_configs/xgb_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/configs/model_configs/xgb_config.py -------------------------------------------------------------------------------- /src/configs/wandb_config.py: -------------------------------------------------------------------------------- 1 | wandb_id = "leogrin" # Enter your ID -------------------------------------------------------------------------------- /src/create_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/create_models.py -------------------------------------------------------------------------------- /src/data_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/data_transforms.py -------------------------------------------------------------------------------- /src/export_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/export_data.py -------------------------------------------------------------------------------- /src/generate_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/generate_data.py -------------------------------------------------------------------------------- /src/generate_dataset_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/generate_dataset_pipeline.py -------------------------------------------------------------------------------- /src/launch_benchmarks/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/launch_benchmarks/README.md -------------------------------------------------------------------------------- /src/launch_benchmarks/launch_agent.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/launch_benchmarks/launch_agent.sh -------------------------------------------------------------------------------- /src/launch_benchmarks/launch_agent_gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/launch_benchmarks/launch_agent_gpu.sh -------------------------------------------------------------------------------- /src/launch_benchmarks/launch_benchmarks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/launch_benchmarks/launch_benchmarks.py -------------------------------------------------------------------------------- /src/launch_benchmarks/launch_on_cluster.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/launch_benchmarks/launch_on_cluster.py -------------------------------------------------------------------------------- /src/launch_benchmarks/launch_xps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/launch_benchmarks/launch_xps.py -------------------------------------------------------------------------------- /src/launch_benchmarks/results/file: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/launch_benchmarks/sweeps/benchmark_sweeps.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/launch_benchmarks/sweeps/benchmark_sweeps.csv -------------------------------------------------------------------------------- /src/launch_benchmarks/sweeps/xps_sweeps.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/launch_benchmarks/sweeps/xps_sweeps.csv -------------------------------------------------------------------------------- /src/launch_benchmarks/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/launch_benchmarks/utils.py -------------------------------------------------------------------------------- /src/models/TabSurvey/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/.gitignore -------------------------------------------------------------------------------- /src/models/TabSurvey/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/Dockerfile -------------------------------------------------------------------------------- /src/models/TabSurvey/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/LICENSE -------------------------------------------------------------------------------- /src/models/TabSurvey/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/README.md -------------------------------------------------------------------------------- /src/models/TabSurvey/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/TabSurvey/attributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/attributions.py -------------------------------------------------------------------------------- /src/models/TabSurvey/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/evaluate.py -------------------------------------------------------------------------------- /src/models/TabSurvey/insights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/insights.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/__init__.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/baseline_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/baseline_models.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/basemodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/basemodel.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/basemodel_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/basemodel_torch.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/danet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/danet.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/danet_lib/DAN_Task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/danet_lib/DAN_Task.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/danet_lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/TabSurvey/models/danet_lib/abstract_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/danet_lib/abstract_model.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/danet_lib/lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/TabSurvey/models/danet_lib/lib/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/danet_lib/lib/callbacks.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/danet_lib/lib/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/danet_lib/lib/logger.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/danet_lib/lib/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/danet_lib/lib/metrics.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/danet_lib/lib/multiclass_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/danet_lib/lib/multiclass_utils.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/danet_lib/lib/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/danet_lib/lib/utils.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/danet_lib/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/danet_lib/main.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/danet_lib/model/AcceleratedModule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/danet_lib/model/AcceleratedModule.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/danet_lib/model/DANet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/danet_lib/model/DANet.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/danet_lib/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/TabSurvey/models/danet_lib/model/sparsemax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/danet_lib/model/sparsemax.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/danet_lib/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/danet_lib/predict.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepfm.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepfm_lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepfm_lib/__init__.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepfm_lib/inputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepfm_lib/inputs.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepfm_lib/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepfm_lib/layers/__init__.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepfm_lib/layers/activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepfm_lib/layers/activation.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepfm_lib/layers/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepfm_lib/layers/core.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepfm_lib/layers/interaction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepfm_lib/layers/interaction.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepfm_lib/layers/sequence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepfm_lib/layers/sequence.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepfm_lib/layers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepfm_lib/layers/utils.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepfm_lib/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .deepfm import DeepFM 2 | -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepfm_lib/models/basemodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepfm_lib/models/basemodel.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepfm_lib/models/deepfm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepfm_lib/models/deepfm.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepgbm.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm_lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepgbm_lib/__init__.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm_lib/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepgbm_lib/config.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm_lib/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepgbm_lib/main.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm_lib/models/CatNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepgbm_lib/models/CatNN.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm_lib/models/DeepGBM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepgbm_lib/models/DeepGBM.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm_lib/models/EmbeddingModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepgbm_lib/models/EmbeddingModel.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm_lib/models/GBDT2NN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepgbm_lib/models/GBDT2NN.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm_lib/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepgbm_lib/models/__init__.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm_lib/preprocess/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm_lib/preprocess/preprocessing_cat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepgbm_lib/preprocess/preprocessing_cat.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm_lib/preprocess/preprocessing_num.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepgbm_lib/preprocess/preprocessing_num.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm_lib/trainModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepgbm_lib/trainModel.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm_lib/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepgbm_lib/utils/__init__.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm_lib/utils/gbdt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepgbm_lib/utils/gbdt.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm_lib/utils/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepgbm_lib/utils/helper.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/deepgbm_lib/utils/tree_model_interpreter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/deepgbm_lib/utils/tree_model_interpreter.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/dnf.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/DNFNetComponents.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/DNFNetComponents.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/model1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/model1.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/model100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/model100.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/model101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/model101.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/model2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/model2.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/model3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/model3.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/model4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/model4.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/model5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/model5.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/model6.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/dnf_lib/DNFNet/DNFNetModels/model6.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/DNFNet/ModelHandler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/dnf_lib/DNFNet/ModelHandler.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/DNFNet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/Utils/NumpyGenerator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/dnf_lib/Utils/NumpyGenerator.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/Utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/Utils/experiment_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/dnf_lib/Utils/experiment_utils.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/Utils/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/dnf_lib/Utils/file_utils.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/TabSurvey/models/dnf_lib/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/dnf_lib/config.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/mlp.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/modeltree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/modeltree.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/neural_additive_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/neural_additive_models.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/node.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/node_lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/node_lib/__init__.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/node_lib/arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/node_lib/arch.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/node_lib/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/node_lib/data.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/node_lib/nn_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/node_lib/nn_utils.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/node_lib/odst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/node_lib/odst.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/node_lib/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/node_lib/trainer.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/node_lib/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/node_lib/utils.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/rln.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/rln.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/saint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/saint.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/saint_lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/TabSurvey/models/saint_lib/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/saint_lib/augmentations.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/saint_lib/data_openml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/saint_lib/data_openml.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/saint_lib/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/saint_lib/models/__init__.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/saint_lib/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/saint_lib/models/model.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/saint_lib/models/pretrainmodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/saint_lib/models/pretrainmodel.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/saint_lib/models/pretrainmodel_vision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/saint_lib/models/pretrainmodel_vision.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/stg_lib/__init__.py: -------------------------------------------------------------------------------- 1 | from .stg import STG -------------------------------------------------------------------------------- /src/models/TabSurvey/models/stg_lib/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/stg_lib/io.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/stg_lib/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/stg_lib/layers.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/stg_lib/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/stg_lib/losses.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/stg_lib/matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/stg_lib/matching.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/stg_lib/meter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/stg_lib/meter.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/stg_lib/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/stg_lib/models.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/stg_lib/stg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/stg_lib/stg.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/stg_lib/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/stg_lib/utils.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/stochastic_gates.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/stochastic_gates.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/tabnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/tabnet.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/tabtransformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/tabtransformer.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/tree_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/tree_models.py -------------------------------------------------------------------------------- /src/models/TabSurvey/models/vime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/models/vime.py -------------------------------------------------------------------------------- /src/models/TabSurvey/testall.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/testall.sh -------------------------------------------------------------------------------- /src/models/TabSurvey/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/train.py -------------------------------------------------------------------------------- /src/models/TabSurvey/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/TabSurvey/utils/baseline_attributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/utils/baseline_attributions.py -------------------------------------------------------------------------------- /src/models/TabSurvey/utils/io_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/utils/io_utils.py -------------------------------------------------------------------------------- /src/models/TabSurvey/utils/load_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/utils/load_data.py -------------------------------------------------------------------------------- /src/models/TabSurvey/utils/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/utils/parser.py -------------------------------------------------------------------------------- /src/models/TabSurvey/utils/scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/utils/scorer.py -------------------------------------------------------------------------------- /src/models/TabSurvey/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/TabSurvey/utils/timer.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/skorch_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/skorch_models.py -------------------------------------------------------------------------------- /src/models/skorch_models_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/skorch_models_regression.py -------------------------------------------------------------------------------- /src/models/tabular/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/.gitignore -------------------------------------------------------------------------------- /src/models/tabular/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/LICENSE -------------------------------------------------------------------------------- /src/models/tabular/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/README.md -------------------------------------------------------------------------------- /src/models/tabular/bin/analysis_ft_transformer_vs_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/analysis_ft_transformer_vs_resnet.py -------------------------------------------------------------------------------- /src/models/tabular/bin/autoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/autoint.py -------------------------------------------------------------------------------- /src/models/tabular/bin/catboost_.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/catboost_.py -------------------------------------------------------------------------------- /src/models/tabular/bin/create_synthetic_data_plots.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/create_synthetic_data_plots.ipynb -------------------------------------------------------------------------------- /src/models/tabular/bin/dcn2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/dcn2.py -------------------------------------------------------------------------------- /src/models/tabular/bin/ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/ensemble.py -------------------------------------------------------------------------------- /src/models/tabular/bin/ft_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/ft_transformer.py -------------------------------------------------------------------------------- /src/models/tabular/bin/grownet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/grownet.py -------------------------------------------------------------------------------- /src/models/tabular/bin/lightgbm_.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/lightgbm_.py -------------------------------------------------------------------------------- /src/models/tabular/bin/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/mlp.py -------------------------------------------------------------------------------- /src/models/tabular/bin/node.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/node.py -------------------------------------------------------------------------------- /src/models/tabular/bin/report.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/report.ipynb -------------------------------------------------------------------------------- /src/models/tabular/bin/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/resnet.py -------------------------------------------------------------------------------- /src/models/tabular/bin/snn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/snn.py -------------------------------------------------------------------------------- /src/models/tabular/bin/tabnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/tabnet.py -------------------------------------------------------------------------------- /src/models/tabular/bin/tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/tune.py -------------------------------------------------------------------------------- /src/models/tabular/bin/xgboost_.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/bin/xgboost_.py -------------------------------------------------------------------------------- /src/models/tabular/lib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/lib/__init__.py -------------------------------------------------------------------------------- /src/models/tabular/lib/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/lib/data.py -------------------------------------------------------------------------------- /src/models/tabular/lib/deep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/lib/deep.py -------------------------------------------------------------------------------- /src/models/tabular/lib/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/lib/env.py -------------------------------------------------------------------------------- /src/models/tabular/lib/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/lib/metrics.py -------------------------------------------------------------------------------- /src/models/tabular/lib/node/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/lib/node/__init__.py -------------------------------------------------------------------------------- /src/models/tabular/lib/node/arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/lib/node/arch.py -------------------------------------------------------------------------------- /src/models/tabular/lib/node/nn_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/lib/node/nn_utils.py -------------------------------------------------------------------------------- /src/models/tabular/lib/node/odst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/lib/node/odst.py -------------------------------------------------------------------------------- /src/models/tabular/lib/node/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/lib/node/utils.py -------------------------------------------------------------------------------- /src/models/tabular/lib/synthetic_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/lib/synthetic_data.py -------------------------------------------------------------------------------- /src/models/tabular/lib/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/lib/util.py -------------------------------------------------------------------------------- /src/models/tabular/pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/pyproject.toml -------------------------------------------------------------------------------- /src/models/tabular/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/requirements.txt -------------------------------------------------------------------------------- /src/models/tabular/setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/models/tabular/setup.cfg -------------------------------------------------------------------------------- /src/preprocessing/preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/preprocessing/preprocessing.py -------------------------------------------------------------------------------- /src/preprocessing/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/preprocessing/utils.py -------------------------------------------------------------------------------- /src/run_experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/run_experiment.py -------------------------------------------------------------------------------- /src/target_function_classif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/target_function_classif.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/train.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/keyword_to_function_conversion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/utils/keyword_to_function_conversion.py -------------------------------------------------------------------------------- /src/utils/skorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/utils/skorch_utils.py -------------------------------------------------------------------------------- /src/utils/tree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/utils/tree.py -------------------------------------------------------------------------------- /src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/src/utils/utils.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/tests/test_models.py -------------------------------------------------------------------------------- /tests/test_tree_quantiles.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LeoGrin/tabular-benchmark/HEAD/tests/test_tree_quantiles.py --------------------------------------------------------------------------------