Java 类名：com.alibaba.alink.operator.batch.feature.TreeModelEncoderBatchOp
Python 类名：TreeModelEncoderBatchOp

功能介绍

使用树模型，将输入数据编码为特征。

参数说明

名称 中文名称 描述 类型 是否必须？ 取值范围 默认值
predictionCol 预测结果列名 预测结果列名 String ✓
modelFilePath 模型的文件路径 模型的文件路径 String null
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.0, "A", 0, 0, 0],
  9.     [2.0, "B", 1, 1, 0],
  10.     [3.0, "C", 2, 2, 1],
  11.     [4.0, "D", 3, 3, 1]
  12. ])
  14. batchSource = BatchOperator.fromDataframe(
  15.     df, schemaStr='f0 double, f1 string, f2 int, f3 int, label int')
  16. streamSource = StreamOperator.fromDataframe(
  17.     df, schemaStr='f0 double, f1 string, f2 int, f3 int, label int')
  19. gbdtTrainOp = GbdtTrainBatchOp()\
  20.     .setFeatureCols(['f0', 'f1', 'f2', 'f3'])\
  21.     .setLabelCol('label')\
  22.     .linkFrom(batchSource)
  24. encoderBatchOp = TreeModelEncoderBatchOp()\
  25.     .setPredictionCol("encoded_features")
  26. encoderStreamOp = TreeModelEncoderStreamOp(gbdtTrainOp)\
  27.     .setPredictionCol("encoded_features")
  29. encoderBatchOp.linkFrom(gbdtTrainOp, batchSource).print()
  30. encoderStreamOp.linkFrom(streamSource).print()
  32. 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.feature.TreeModelEncoderBatchOp;
  5. import com.alibaba.alink.operator.batch.source.MemSourceBatchOp;
  6. import com.alibaba.alink.operator.stream.StreamOperator;
  7. import com.alibaba.alink.operator.stream.feature.TreeModelEncoderStreamOp;
  8. import com.alibaba.alink.operator.stream.source.MemSourceStreamOp;
  10. import com.alibaba.alink.operator.batch.classification.GbdtTrainBatchOp;
  12. import org.junit.Test;
  14. import java.util.Arrays;
  15. import java.util.List;
  17. public class TreeModelEncoderBatchOpTest {
  18.     @Test
  19.     public void testTreeModelEncoderBatchOp() throws Exception {
  20.         List <Row> df = Arrays.asList(
  21.             Row.of(1.0, "A", 0, 0, 0),
  22.             Row.of(2.0, "B", 1, 1, 0),
  23.             Row.of(3.0, "C", 2, 2, 1),
  24.             Row.of(4.0, "D", 3, 3, 1)
  25.         );
  27.         BatchOperator batchSource = new MemSourceBatchOp(
  28.             df, "f0 double, f1 string, f2 int, f3 int, label int");
  29.         StreamOperator streamSource = new MemSourceStreamOp(
  30.             df, "f0 double, f1 string, f2 int, f3 int, label int");
  32.         BatchOperator gbdtTrainOp = new GbdtTrainBatchOp()
  33.             .setFeatureCols(new String[] {"f0", "f1", "f2", "f3"})
  34.             .setLabelCol("label")
  35.             .linkFrom(batchSource);
  37.         BatchOperator encoderBatchOp = new TreeModelEncoderBatchOp()
  38.             .setPredictionCol("encoded_features");
  39.         StreamOperator encoderStreamOp =new TreeModelEncoderStreamOp(gbdtTrainOp)
  40.             .setPredictionCol("encoded_features");
  42.         encoderBatchOp.linkFrom(gbdtTrainOp, batchSource).print();
  43.         encoderStreamOp.linkFrom(streamSource).print();
  45.         StreamOperator.execute();
  46.     }
  47. }

运行结果

| f0 | f1 | f2 | f3 | label | encoded_features | | —- | —- | —- | —- | —- | —- |

| 1.0000 | A | 0 | 0 | 0 | $123$0:1.0 2:1.0 4:1.0 6:1.0 8:1.0 10:1.0 12:1.0 14:1.0 16:1.0 18:1.0 20:1.0 22:1.0 24:1.0 26:1.0 28:1.0 30:1.0 32:1.0 34:1.0 36:1.0 38:1.0 40:1.0 42:1.0 44:1.0 46:1.0 47:1.0 48:1.0 49:1.0 50:1.0 51:1.0 52:1.0 53:1.0 54:1.0 55:1.0 56:1.0 57:1.0 58:1.0 59:1.0 60:1.0 61:1.0 62:1.0 63:1.0 64:1.0 65:1.0 66:1.0 67:1.0 68:1.0 69:1.0 70:1.0 71:1.0 72:1.0 73:1.0 74:1.0 75:1.0 76:1.0 77:1.0 78:1.0 79:1.0 80:1.0 81:1.0 82:1.0 83:1.0 84:1.0 85:1.0 86:1.0 87:1.0 88:1.0 89:1.0 90:1.0 91:1.0 92:1.0 93:1.0 94:1.0 95:1.0 96:1.0 97:1.0 98:1.0 99:1.0 100:1.0 101:1.0 102:1.0 103:1.0 104:1.0 105:1.0 106:1.0 107:1.0 108:1.0 109:1.0 110:1.0 111:1.0 112:1.0 113:1.0 114:1.0 115:1.0 116:1.0 117:1.0 118:1.0 119:1.0 120:1.0 121:1.0 122:1.0 |

| 2.0000 | B | 1 | 1 | 0 | $123$0:1.0 2:1.0 4:1.0 6:1.0 8:1.0 10:1.0 12:1.0 14:1.0 16:1.0 18:1.0 20:1.0 22:1.0 24:1.0 26:1.0 28:1.0 30:1.0 32:1.0 34:1.0 36:1.0 38:1.0 40:1.0 42:1.0 44:1.0 46:1.0 47:1.0 48:1.0 49:1.0 50:1.0 51:1.0 52:1.0 53:1.0 54:1.0 55:1.0 56:1.0 57:1.0 58:1.0 59:1.0 60:1.0 61:1.0 62:1.0 63:1.0 64:1.0 65:1.0 66:1.0 67:1.0 68:1.0 69:1.0 70:1.0 71:1.0 72:1.0 73:1.0 74:1.0 75:1.0 76:1.0 77:1.0 78:1.0 79:1.0 80:1.0 81:1.0 82:1.0 83:1.0 84:1.0 85:1.0 86:1.0 87:1.0 88:1.0 89:1.0 90:1.0 91:1.0 92:1.0 93:1.0 94:1.0 95:1.0 96:1.0 97:1.0 98:1.0 99:1.0 100:1.0 101:1.0 102:1.0 103:1.0 104:1.0 105:1.0 106:1.0 107:1.0 108:1.0 109:1.0 110:1.0 111:1.0 112:1.0 113:1.0 114:1.0 115:1.0 116:1.0 117:1.0 118:1.0 119:1.0 120:1.0 121:1.0 122:1.0 |

| 3.0000 | C | 2 | 2 | 1 | $123$1:1.0 3:1.0 5:1.0 7:1.0 9:1.0 11:1.0 13:1.0 15:1.0 17:1.0 19:1.0 21:1.0 23:1.0 25:1.0 27:1.0 29:1.0 31:1.0 33:1.0 35:1.0 37:1.0 39:1.0 41:1.0 43:1.0 45:1.0 46:1.0 47:1.0 48:1.0 49:1.0 50:1.0 51:1.0 52:1.0 53:1.0 54:1.0 55:1.0 56:1.0 57:1.0 58:1.0 59:1.0 60:1.0 61:1.0 62:1.0 63:1.0 64:1.0 65:1.0 66:1.0 67:1.0 68:1.0 69:1.0 70:1.0 71:1.0 72:1.0 73:1.0 74:1.0 75:1.0 76:1.0 77:1.0 78:1.0 79:1.0 80:1.0 81:1.0 82:1.0 83:1.0 84:1.0 85:1.0 86:1.0 87:1.0 88:1.0 89:1.0 90:1.0 91:1.0 92:1.0 93:1.0 94:1.0 95:1.0 96:1.0 97:1.0 98:1.0 99:1.0 100:1.0 101:1.0 102:1.0 103:1.0 104:1.0 105:1.0 106:1.0 107:1.0 108:1.0 109:1.0 110:1.0 111:1.0 112:1.0 113:1.0 114:1.0 115:1.0 116:1.0 117:1.0 118:1.0 119:1.0 120:1.0 121:1.0 122:1.0 |

| 4.0000 | D | 3 | 3 | 1 | $123$1:1.0 3:1.0 5:1.0 7:1.0 9:1.0 11:1.0 13:1.0 15:1.0 17:1.0 19:1.0 21:1.0 23:1.0 25:1.0 27:1.0 29:1.0 31:1.0 33:1.0 35:1.0 37:1.0 39:1.0 41:1.0 43:1.0 45:1.0 46:1.0 47:1.0 48:1.0 49:1.0 50:1.0 51:1.0 52:1.0 53:1.0 54:1.0 55:1.0 56:1.0 57:1.0 58:1.0 59:1.0 60:1.0 61:1.0 62:1.0 63:1.0 64:1.0 65:1.0 66:1.0 67:1.0 68:1.0 69:1.0 70:1.0 71:1.0 72:1.0 73:1.0 74:1.0 75:1.0 76:1.0 77:1.0 78:1.0 79:1.0 80:1.0 81:1.0 82:1.0 83:1.0 84:1.0 85:1.0 86:1.0 87:1.0 88:1.0 89:1.0 90:1.0 91:1.0 92:1.0 93:1.0 94:1.0 95:1.0 96:1.0 97:1.0 98:1.0 99:1.0 100:1.0 101:1.0 102:1.0 103:1.0 104:1.0 105:1.0 106:1.0 107:1.0 108:1.0 109:1.0 110:1.0 111:1.0 112:1.0 113:1.0 114:1.0 115:1.0 116:1.0 117:1.0 118:1.0 119:1.0 120:1.0 121:1.0 122:1.0 |