TPOT2: A New Graph-Based Implementation of the Tree-Based Pipeline Optimization Tool for Automated Machine Learning
摘要
In this chapter, we present a new implementation of the popularPipeline optimization Tree-Based Pipeline Optimization Tool (TPOT). This new implementation, called TPOT2, was rebuilt from the ground up to be more modular, easier to maintain, and easier to expand. TPOT2 comes with new features and optimizationsOptimization, such as a more flexible graph-based representation of Scikit-Learn pipelines and the ability to specify various aspects of the evolutionary run. Using experiments on multiple benchmarkBenchmark datasets, we show that TPOT2 performs at least as well as TPOT1 with equivalent settings, with stronger performance on a few datasets. We outline some future directions for further optimizations and applications.