search

iamDecode / sklearn-pmml-model

A library to parse and convert PMML models into Scikit-learn estimators.
BSD 2-Clause "Simplified" License
76 stars 15 forks source link

Support regression for tree-based models #25

Closed iamDecode closed 3 years ago

iamDecode commented 3 years ago

As gradient boosting uses regression trees internally, #23 already contained the majority of preparation for supporting regression in tree-based models. This PR adds two new classes PMMLTreeRegressor (extending DecisionTreeRegressor) and PMMLForestRegressor (extending RandomForestRegressor) for regression problems.

codecov[bot] commented 3 years ago

Codecov Report

Merging #25 (84f380b) into master (afb42d0) will not change coverage. The diff coverage is 100.00%.

Impacted file tree graph

@@            Coverage Diff             @@
##            master       #25    +/-   ##
==========================================
  Coverage   100.00%   100.00%            
==========================================
  Files           12        12            
  Lines          624       729   +105     
==========================================
+ Hits           624       729   +105
Impacted Files Coverage Δ
sklearn_pmml_model/ensemble/__init__.py 100.00% <100.00%> (ø)
sklearn_pmml_model/ensemble/forest.py 100.00% <100.00%> (ø)
sklearn_pmml_model/ensemble/gb.py 100.00% <100.00%> (ø)
sklearn_pmml_model/tree/__init__.py 100.00% <100.00%> (ø)
sklearn_pmml_model/tree/tree.py 100.00% <100.00%> (ø)

Continue to review full report at Codecov.

Legend - Click here to learn more Δ = absolute <relative> (impact), ø = not affected, ? = missing data Powered by Codecov. Last update afb42d0...84f380b. Read the comment docs.