Java 类名：com.alibaba.alink.operator.stream.dataproc.vector.VectorBiFunctionStreamOp
Python 类名：VectorBiFunctionStreamOp

功能介绍

  • 对两个向量进行操作的函数，支持minus(减),plus(加),dot(内积),merge(拼接),EuclidDistance(欧式距离),Cosine(cos值), ElementWiseMultiply(点乘)。
  • 支持稀疏和稠密两种 Vector。

    参数说明

    | 名称 | 中文名称 | 描述 | 类型 | 是否必须？ | 取值范围 | 默认值 | | —- | —- | —- | —- | —- | —- | —- |

| biFuncName | 函数名字 | 函数操作名称, 可取minus(减),plus(加),dot(内积),merge(拼接),EuclidDistance(欧式距离),Cosine(cos值), ElementWiseMultiply(点乘). | String | ✓ | “Minus”, “Dot”, “Plus”, “Merge”, “EuclidDistance”, “Cosine”, “ElementWiseMultiply” | |

| outputCol | 输出结果列列名 | 输出结果列列名，必选 | String | ✓ | | |

| selectedCols | 选择的列名 | 计算列对应的列名列表 | String[] | ✓ | | |

| reservedCols | 算法保留列名 | 算法保留列 | String[] | | | null |

| numThreads | 组件多线程线程个数 | 组件多线程线程个数 | Integer | | | 1 |

代码示例

Python 代码

  1. from pyalink.alink import *
  3. import pandas as pd
  5. useLocalEnv(1)
  7. df = pd.DataFrame([
  8.     ["1 2 3", "2 3 4"]
  9. ])
  10. data = StreamOperator.fromDataframe(df, schemaStr="vec1 string, vec2 string")
  11. VectorBiFunctionStreamOp() \
  12.         .setSelectedCols(["vec1", "vec2"]) \
  13.         .setBiFuncName("minus").setOutputCol("vec_minus").linkFrom(data).print();
  14. StreamOperator.execute()

Java 代码

  1. import org.apache.flink.types.Row;
  3. import com.alibaba.alink.operator.batch.BatchOperator;
  4. import com.alibaba.alink.operator.batch.source.MemSourceBatchOp;
  5. import com.alibaba.alink.pipeline.dataproc.vector.VectorBiFunction;
  6. import com.alibaba.alink.testutil.AlinkTestBase;
  7. import org.junit.Test;
  9. import java.util.ArrayList;
  10. import java.util.List;
  12. public class ToVectorBiFunctionTest extends AlinkTestBase {
  13.     @Test
  14.     public void testBiVectorFunction() throws Exception {
  15.         List <Row> df = new ArrayList <>();
  16.         df.add(Row.of("1 2 3", "2 3 4"));
  17.         StreamOperator <?> data = new MemSourceStreamOp(df, "vec1 string, vec2 string");
  18.         new VectorBiFunctionStreamOp()
  19.             .setSelectedCols("vec1", "vec2")
  20.             .setBiFuncName("minus")
  21.             .setOutputCol("vec_minus").linkFrom(data).print();
  22.         StreamOperator.execute();
  23.     }
  24. }

运行结果

| vec1 | vec2 | vec_minus | | —- | —- | —- |

| 1 2 3 | 2 3 4 | -1.0 -1.0 -1.0 |