Java 类名:com.alibaba.alink.operator.stream.sink.LibSvmSinkStreamOp
Python 类名:LibSvmSinkStreamOp

功能介绍

写出LibSvm格式文件,支持写出到本地文件和HDFS文件。

参数说明

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

| filePath | 文件路径 | 文件路径 | String | ✓ | | |

| labelCol | 标签列名 | 输入表中的标签列名 | String | ✓ | | |

| vectorCol | 向量列名 | 向量列对应的列名 | String | ✓ | | |

| overwriteSink | 是否覆写已有数据 | 是否覆写已有数据 | Boolean | | | false |

| startIndex | 起始索引 | 起始索引 | Integer | | | 1 |

代码示例

Python 代码

  1. from pyalink.alink import *
  2. import pandas as pd
  3. useLocalEnv(1)
  4. df_data = pd.DataFrame([
  5. ['1:2.0 2:1.0 4:0.5', 1.5],
  6. ['1:2.0 2:1.0 4:0.5', 1.7],
  7. ['1:2.0 2:1.0 4:0.5', 3.6]
  8. ])
  9. stream_data = StreamOperator.fromDataframe(df_data, schemaStr='f1 string, f2 double')
  10. sink = LibSvmSinkStreamOp().setFilePath('/tmp/abc.svm').setLabelCol("f2").setVectorCol("f1").setOverwriteSink(True)
  11. stream_data = stream_data.link(sink)
  12. StreamOperator.execute()

Java 代码

  1. import org.apache.flink.types.Row;
  2. import com.alibaba.alink.operator.stream.StreamOperator;
  3. import com.alibaba.alink.operator.stream.sink.LibSvmSinkStreamOp;
  4. import com.alibaba.alink.operator.stream.source.MemSourceStreamOp;
  5. import org.junit.Test;
  6. import java.util.Arrays;
  7. import java.util.List;
  8. public class LibSvmSinkStreamOpTest {
  9. @Test
  10. public void testLibSvmSinkStreamOp() throws Exception {
  11. List <Row> df_data = Arrays.asList(
  12. Row.of("1:2.0 2:1.0 4:0.5", 1.5),
  13. Row.of("1:2.0 2:1.0 4:0.5", 1.7),
  14. Row.of("1:2.0 2:1.0 4:0.5", 3.6)
  15. );
  16. StreamOperator <?> stream_data = new MemSourceStreamOp(df_data, "f1 string, f2 double");
  17. StreamOperator <?> sink = new LibSvmSinkStreamOp().setFilePath("/tmp/abc.svm").setLabelCol("f2").setVectorCol(
  18. "f1").setOverwriteSink(true);
  19. stream_data = stream_data.link(sink);
  20. StreamOperator.execute();
  21. }
  22. }