├── .github └── workflows │ └── release.yml ├── .gitignore ├── AGENT_GUIDE.md ├── LICENSE ├── MANIFEST.in ├── README.md ├── examples └── Spiral Dataset.ipynb ├── pyproject.toml ├── setup.py ├── tests ├── __init__.py ├── full_numerai_test.py ├── numerai_test.py ├── test_comparison_lightgbm.py ├── test_feature_importance.py ├── test_fit_predict_corr.py ├── test_invariant.py ├── test_multiclass.py ├── test_random_state.py └── test_warm_start.py ├── version.txt └── warpgbm ├── __init__.py ├── core.py ├── cuda ├── __init__.py ├── best_split_kernel.cu ├── binner.cu ├── histogram_kernel.cu ├── node_kernel.cpp └── predict.cu └── metrics.py /.github/workflows/release.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/.github/workflows/release.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/.gitignore -------------------------------------------------------------------------------- /AGENT_GUIDE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/AGENT_GUIDE.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/README.md -------------------------------------------------------------------------------- /examples/Spiral Dataset.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/examples/Spiral Dataset.ipynb -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/pyproject.toml -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/full_numerai_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/tests/full_numerai_test.py -------------------------------------------------------------------------------- /tests/numerai_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/tests/numerai_test.py -------------------------------------------------------------------------------- /tests/test_comparison_lightgbm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/tests/test_comparison_lightgbm.py -------------------------------------------------------------------------------- /tests/test_feature_importance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/tests/test_feature_importance.py -------------------------------------------------------------------------------- /tests/test_fit_predict_corr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/tests/test_fit_predict_corr.py -------------------------------------------------------------------------------- /tests/test_invariant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/tests/test_invariant.py -------------------------------------------------------------------------------- /tests/test_multiclass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/tests/test_multiclass.py -------------------------------------------------------------------------------- /tests/test_random_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/tests/test_random_state.py -------------------------------------------------------------------------------- /tests/test_warm_start.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/tests/test_warm_start.py -------------------------------------------------------------------------------- /version.txt: -------------------------------------------------------------------------------- 1 | 2.2.0 2 | -------------------------------------------------------------------------------- /warpgbm/__init__.py: -------------------------------------------------------------------------------- 1 | from .core import WarpGBM -------------------------------------------------------------------------------- /warpgbm/core.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/warpgbm/core.py -------------------------------------------------------------------------------- /warpgbm/cuda/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /warpgbm/cuda/best_split_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/warpgbm/cuda/best_split_kernel.cu -------------------------------------------------------------------------------- /warpgbm/cuda/binner.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/warpgbm/cuda/binner.cu -------------------------------------------------------------------------------- /warpgbm/cuda/histogram_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/warpgbm/cuda/histogram_kernel.cu -------------------------------------------------------------------------------- /warpgbm/cuda/node_kernel.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/warpgbm/cuda/node_kernel.cpp -------------------------------------------------------------------------------- /warpgbm/cuda/predict.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/warpgbm/cuda/predict.cu -------------------------------------------------------------------------------- /warpgbm/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jefferythewind/warpgbm/HEAD/warpgbm/metrics.py --------------------------------------------------------------------------------