LibSvm文件导出 (LibSvmSinkStreamOp)

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 代码

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

Java 代码

import org.apache.flink.types.Row;
import com.alibaba.alink.operator.stream.StreamOperator;
import com.alibaba.alink.operator.stream.sink.LibSvmSinkStreamOp;
import com.alibaba.alink.operator.stream.source.MemSourceStreamOp;
import org.junit.Test;
import java.util.Arrays;
import java.util.List;
public class LibSvmSinkStreamOpTest {
    @Test
    public void testLibSvmSinkStreamOp() throws Exception {
        List <Row> df_data = Arrays.asList(
            Row.of("1:2.0 2:1.0 4:0.5", 1.5),
            Row.of("1:2.0 2:1.0 4:0.5", 1.7),
            Row.of("1:2.0 2:1.0 4:0.5", 3.6)
        );
        StreamOperator <?> stream_data = new MemSourceStreamOp(df_data, "f1 string, f2  double");
        StreamOperator <?> sink = new LibSvmSinkStreamOp().setFilePath("/tmp/abc.svm").setLabelCol("f2").setVectorCol(
            "f1").setOverwriteSink(true);
        stream_data = stream_data.link(sink);
        StreamOperator.execute();
    }
}