Todo Add test cases Add README.md Add Example Notebook Add Static Typing (mypy) Add function for xgboost and lightGBM