EpistasisLab / tpot

A Python Automated Machine Learning tool that optimizes machine learning pipelines using genetic programming.
http://epistasislab.github.io/tpot/
GNU Lesser General Public License v3.0
9.66k stars 1.56k forks source link

AttributeError: 'Pipeline' object has no attribute 'pred_dist' #1085

Open astrogilda opened 4 years ago

astrogilda commented 4 years ago

Hey!

Let me begin by thanking y'all for a fantastic package!! I was wanting to use tpot for finding a good sklearn pipeline with ngboost as the final estimator, for a regression problem with one output variable. However, since the probabilistic prediction method for ngboost is called 'pred_dist', which is not something sklearn supports, python throws an error. Do you have any suggestions?

weixuanfu commented 4 years ago

Could you please provide a small demo with Boston benchmark and config_dict of ngboost to reproduce this issue?