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

** 以下代码仅用于示意，可能需要修改部分代码或者配置环境后才能正常运行！**

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();
	}
}