Java 类名：com.alibaba.alink.pipeline.dataproc.vector.VectorInteraction
Python 类名：VectorInteraction

功能介绍

对两个vector 中的元素两两相乘，并组成一个新的向量。

参数说明

名称 中文名称 描述 类型 是否必须？ 取值范围 默认值
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.     ["$8$1:3,2:4,4:7", "$8$1:3,2:4,4:7"],
  9.     ["$8$0:3,5:5", "$8$1:2,2:4,4:7"],
  10.     ["$8$2:4,4:5", "$8$1:3,2:3,4:7"]
  11. ])
  12. data = BatchOperator.fromDataframe(df, schemaStr="vec1 string, vec2 string")
  13. vecInter = VectorInteraction().setSelectedCols(["vec1","vec2"]).setOutputCol("vec_product")
  14. vecInter.transform(data).collectToDataframe()

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.VectorInteraction;
  6. import org.junit.Test;
  8. import java.util.Arrays;
  9. import java.util.List;
  11. public class VectorInteractionTest {
  12.     @Test
  13.     public void testVectorInteraction() throws Exception {
  14.         List <Row> df = Arrays.asList(
  15.             Row.of("$8$1:3,2:4,4:7", "$8$1:3,2:4,4:7"),
  16.             Row.of("$8$0:3,5:5", "$8$1:2,2:4,4:7"),
  17.             Row.of("$8$2:4,4:5", "$8$1:3,2:3,4:7")
  18.         );
  19.         BatchOperator <?> data = new MemSourceBatchOp(df, "vec1 string, vec2 string");
  20.         VectorInteraction vecInter = new VectorInteraction().setSelectedCols("vec1", "vec2").setOutputCol(
  21.             "vec_product");
  22.         vecInter.transform(data).print();
  23.     }
  24. }

运行结果

| vec1 | vec2 | vec_product | | —- | —- | —- |

| $8$1:3,2:4,4:7 | $8$1:3,2:4,4:7 | $64$9:9.0 10:12.0 12:21.0 17:12.0 18:16.0 20:28.0 33:21.0 34:28.0 36:49.0 |

| $8$0:3,5:5 | $8$1:2,2:4,4:7 | $64$8:6.0 13:10.0 16:12.0 21:20.0 32:21.0 37:35.0 |

| $8$2:4,4:5 | $8$1:3,2:3,4:7 | $64$10:12.0 12:15.0 18:12.0 20:15.0 34:28.0 36:35.0 |