MLeap Scikit-Learn 集成

    1. 序列化 Scikit-Learn Pipeline 并在 MLeap Runtime 中执行。
    2. 序列化 Scikit-Learn Pipeline,以及反序列化回 Spark Pipeline。

    如我们之前所说,MLeap Runtime 目前仅提供了 Scala 的函数库,我们会在未来添加 Python 绑定支持,但目前已经完全能够不依赖于 Scikit-Learn 和 Numpy 去执行 Pipeline 了。

    Scikit-Learn Transformer 和 Spark Transformer 的运作存在以下几点重要区别:

    1. 所有的 Spark Transformer 都带有 、opinputCol 以及 outputCol 属性,而 Scikit-Learn 并没有(ihainan: 不会全有,还是全部没有?)。
    2. Spark Transformer 可以作用于一个向量,而 Scikit-Learn Transformer 作用于一个 n 维数组和矩阵。
    3. Spark 是由 Scala 语言编写而成,因此可以非常方便地添加隐式方法和属性,Scikit-Learn 则使用了一点小技巧,要求用户去调用 setattr() 方法。

    由于这三个额外的差异,在使用 MLeap 来扩展 Scikit-Learn 的功能时,需要遵循一些规范。

    首先,我们必须初始化每一个 Transformer,使其包含如下属性:

    • Op: 唯一的 op 名字,用于链接基于 Spark 的 Transformer(例如,Scikit-Learn 和 Spark 中的 Standard Scaler 是等价的,因为我们需要用 opstandard_scaler 来表示这个 Transformer)。
    • Name: 为每一个 Transformer 提供唯一的 name。例如,如果你有多个 Standard Scaler 实例,那么每一个实例都需要提供独一无二的 name。
    • Input Column:严格用在序列化过程,指明输入字段是什么。
    • Output Column:严格用在序列化过程,指明输出字段是什么。

    首先导入所有必要的依赖库:

    1. # Create a pandas DataFrame
    2. df = pd.DataFrame(np.random.randn(10, 5), columns=['a', 'b', 'c', 'd', 'e'])

    让我们定义两个 Transformer,一个为特征提取器(FeatureExtractor),用于提取我们想要进行归一化的字段,以及一个 Standard Scaler,用于执行标准归一化操作。

    现在我们已经定义好了 Transformer,接下来需要把它们组装为 Pipeline,并对 Data Frame 执行转换操作。

    1. # Now let's create a small pipeline using the Feature Extractor and the Standard Scaler
    2. standard_scaler_pipeline = Pipeline([(feature_extractor_tf.name, feature_extractor_tf),
    3. standard_scaler_pipeline.mlinit()
    4. # Now let's run it on our test DataFrame
    5. standard_scaler_pipeline.fit_transform(df)
    6. # Printed output
    7. array([[ 0.2070446 , 0.30612846, -0.91620529],
    8. [ 0.81463009, -0.26668287, 1.95663995],
    9. [-0.43931405, 0.13214763, -0.10700743],
    10. [ 0.43992551, -0.2985418 , -0.89093752],
    11. [-0.15391539, -2.20828471, 0.5361159 ],
    12. [-1.07689244, 1.61019861, 1.42868885],
    13. [ 0.87874789, 1.43146482, -0.44362038],
    14. [-1.60105094, -0.40130005, -0.10754886],
    15. [ 1.87161513, -0.11630878, -1.40990552]])

    我们演示了如何将 Transformer 作用于一系列特征之上,但是这个操作的输出仅仅只是一个 n 维的数组,如果我们需要把它提供如回归模型这样的模型使用的话,还需要将原始的数据与这个输出联合在一起。让我们看下如何使用特征联合来将多个 Transformer 的结果联合在一起。

    首先,我们继续创建一个新的 Transformer:一个 MinMaxScaler,作用于 Data Frame 中剩下的两个特征之上。

    最后,让我们使用特征联合来合并两个 Pipeline。需要注意的是在合并之前,你无需去执行 Pipeline 的 fit 或者 fit_transform 操作。

    1. # Import MLeap extension to Feature Unions
    2. import mleap.sklearn.feature_union
    3. # Import Feature Union
    4. from sklearn.pipeline import FeatureUnion
    5. feature_union = FeatureUnion([
    6. (standard_scaler_pipeline.name, standard_scaler_pipeline),
    7. (min_max_scaler_pipeline.name, min_max_scaler_pipeline)
    8. ])
    9. feature_union.mlinit()
    10. # Create pipeline out of the Feature Union
    11. feature_union_pipeline = Pipeline([(feature_union.name, feature_union)])
    12. feature_union_pipeline.mlinit()
    13. # Execute it on our data frame
    14. feature_union_pipeline.fit_transform(df)
    15. array([[ 0.2070446 , 0.30612846, -0.91620529, 0.58433367, 0.72234095],
    16. [ 0.81463009, -0.26668287, 1.95663995, 0.21145259, 0.72993807],
    17. [-0.94079041, -0.18882131, -0.0462197 , 0.52661493, 0.59771784],
    18. [-0.43931405, 0.13214763, -0.10700743, 0.29403088, 0.19431993],
    19. [ 0.43992551, -0.2985418 , -0.89093752, 0.48838789, 1. ],
    20. [-0.15391539, -2.20828471, 0.5361159 , 1. , 0.46456522],
    21. [-1.07689244, 1.61019861, 1.42868885, 0.36402459, 0.43669119],
    22. [ 0.87874789, 1.43146482, -0.44362038, 0. , 0.74182958],
    23. [ 1.87161513, -0.11630878, -1.40990552, 0.33707035, 0.39792128]])

    For example jar:file:/tmp/mleap-bundle.zip.

    例如: .

    设置 init=Ture告知序列化器我们创建一个 Bundle,而非仅仅只是序列化 Transformer。

    即将支持。

    即将支持。