It would be nice to have table with TPOT vs sklearn operators.
AFAIK not all operators from sklearn are included in tpot. It could be used as:
Good idea. I've added this to the enhancements list.
This is our current coverage:
@teaearlgraycold for someone coming brand new into the project, is there a good example of what needs to be done for the list you just posted?
@westonplatter - here's the code I used
from sklearn import * # Needed to discover all subclasses
import tpot
import sklearn
def all_subclasses(cls):
return cls.__subclasses__() + [g for s in cls.__subclasses__() for g in all_subclasses(s)]
tpot_estimators = set([x.__name__ for x in tpot.operators.Operator.inheritors()])
sklearn_estimators = set([x.__name__ for x in all_subclasses(sklearn.base.BaseEstimator)
if x.__name__[0] != '_' and not x.__name__.startswith("Base")])
for est in sklearn_estimators:
marker = 'X' if est in tpot_estimators else ' '
print("- [{}] {}".format(marker, est))
@teaearlgraycold thanks for the example. I'll take a look.
@westonplatter, that code requires you use the version of TPOT that is currently in the development branch (0.5).
Most helpful comment
This is our current coverage: