Amltk 解析器

该文件是TPOT库的一部分。

当前版本的TPOT是由以下人员在Cedars-Sinai开发的： - Pedro Henrique Ribeiro (https://github.com/perib, https://www.linkedin.com/in/pedro-ribeiro/) - Anil Saini (anil.saini@cshs.org) - Jose Hernandez (jgh9094@gmail.com) - Jay Moran (jay.moran@cshs.org) - Nicholas Matsumoto (nicholas.matsumoto@cshs.org) - Hyunjun Choi (hyunjun.choi@cshs.org) - Miguel E. Hernandez (miguel.e.hernandez@cshs.org) - Jason Moore (moorejh28@gmail.com)

TPOT的原始版本主要由宾夕法尼亚大学的以下人员开发： - Randal S. Olson (rso@randalolson.com) - Weixuan Fu (weixuanf@upenn.edu) - Daniel Angell (dpa34@drexel.edu) - Jason Moore (moorejh28@gmail.com) - 以及许多慷慨的开源贡献者

TPOT 是免费软件：您可以根据自由软件基金会发布的 GNU 宽通用公共许可证的条款重新分发和/或修改它，许可证的版本可以是第 3 版，或者（根据您的选择）任何以后的版本。

TPOT 的发布是希望它能有用， 但没有任何保证；甚至没有对 适销性或特定用途适用性的暗示保证。更多详情请参阅 GNU 较宽松通用公共许可证。

您应该已经收到了一份GNU较宽松通用公共许可证的副本，随TPOT一起提供。如果没有，请参见http://www.gnu.org/licenses/。

 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

def tpot2_parser(
    node: Node,
    ):
    """
    Convert amltk pipeline search space into a tpot2 pipeline search space.

    Parameters
    ----------
    node: amltk.pipeline.Node
        The node to convert.

    Returns
    -------
    tpot2.search_spaces.base.SearchSpace
        The equivalent TPOT search space which can be optimized by TPOT.
    """

    if isinstance(node, Component):
        return component_to_estimatornode(node)
    elif isinstance(node, Sequential):
        return sequential_to_sequentialpipeline(node)
    elif isinstance(node, Choice):
        return choice_to_choicepipeline(node)
    elif isinstance(node, Fixed):
        return fixed_to_estimatornode(node)
    elif isinstance(node, Split):
        return split_to_unionpipeline(node)
    else:
        raise ValueError(f"Node type {type(node)} not supported")