Add Makefile

Makefile

@@ -0,0 +1,28 @@
+SHELL := /bin/bash
+.DEFAULT_GOAL := help
+.PHONY: coverage deps help lint push test doc build
+
+lint:  ## Lint and static-check
+	black src
+	flake8 src
+	mypy src
+
+audit: ## Audit pip
+	pip-audit
+
+help: ## Show help message
+	@IFS=$$'\n' ; \
+	help_lines=(`fgrep -h "##" $(MAKEFILE_LIST) | fgrep -v fgrep | sed -e 's/\\$$//' | sed -e 's/##/:/'`); \
+	printf "%s\n\n" "Usage: make [task]"; \
+	printf "%-20s %s\n" "task" "help" ; \
+	printf "%-20s %s\n" "------" "----" ; \
+	for help_line in $${help_lines[@]}; do \
+		IFS=$$':' ; \
+		help_split=($$help_line) ; \
+		help_command=`echo $${help_split[0]} | sed -e 's/^ *//' -e 's/ *$$//'` ; \
+		help_info=`echo $${help_split[2]} | sed -e 's/^ *//' -e 's/ *$$//'` ; \
+		printf '\033[36m'; \
+		printf "%-20s %s" $$help_command ; \
+		printf '\033[0m'; \
+		printf "%s\n" $$help_info; \
+	done

src/Models.py

@@ -45,7 +45,9 @@ class Models:
             depth = 0
         elif name == "Bagging":
             if hasattr(result.base_estimator_, "nodes_leaves"):
-                nodes, leaves = list(zip(*[x.nodes_leaves() for x in result.estimators_]))
+                nodes, leaves = list(
+                    zip(*[x.nodes_leaves() for x in result.estimators_])
+                )
                 nodes, leaves = mean(nodes), mean(leaves)
                 depth = mean([x.depth_ for x in result.estimators_])
             elif hasattr(result.base_estimator_, "tree_"):