聊聊基于Alink库的特征工程方法
独热编码
OneHotEncoder 是用于将类别型特征转换为独热编码的类。独热编码是一种常用的特征编码方式，特别适用于处理类别型特征，将其转换为数值型特征。
对于每个类别型特征，OneHotEncoder 将其编码成一个长度为类别数量的向量。
每个类别对应一个维度，如果样本的该特征值为该类别，则对应维度置1，其他维度置0。
示例：
List <Row> df = Arrays.asList(
Row.of("a", 1),
Row.of("b", 1),
Row.of("c", 1),
Row.of("e", 2),
Row.of("a", 2),
Row.of("b", 1),
Row.of("c", 2),
Row.of("d", 2),
Row.of(null, 1)
);
BatchOperator <?> inOp = new MemSourceBatchOp(df, "query string, weight int");
OneHotEncoder one_hot = new OneHotEncoder().setSelectedCols("query").setEncode(Encode.VECTOR);
// 或者单独设置输出列
// OneHotEncoder one_hot = new OneHotEncoder().setSelectedCols("query").setOutputCols("output");
one_hot.fit(inOp).transform(inOp).print();输出结果如下所示：
query | weight |
|---|---|
$5$0:1.0 | 1 |
$5$1:1.0 | 1 |
$5$2:1.0 | 1 |
$5$ | 2 |
$5$0:1.0 | 2 |
$5$1:1.0 | 1 |
$5$2:1.0 | 2 |
$5$3:1.0 | 2 |
$5$4:1.0 | 1 |
向量聚合
VectorAssembler 是用于将多列特征合并为单列特征向量的类。它将多个特征列的值合并为一个特征向量，通常用于特征工程的最后阶段，以准备机器学习模型的输入特征。
对于每个样本，VectorAssembler 将选定的特征列的值合并成一个特征向量。特征向量的维度等于选定的特征列数，每个维度对应一个特征列的值。
合并的特征列仅支持数值类型。
示例：
List <Row> df = Arrays.asList(
Row.of("0", "$6$1:2.0 2:3.0 5:4.3", "3.0 2.0 3.0"),
Row.of("1", "$8$1:2.0 2:3.0 7:4.3", "3.0 2.0 3.0"),
Row.of("2", "$8$1:2.0 2:3.0 7:4.3", "2.0 3.0 2.2")
);
MemSourceBatchOp data = new MemSourceBatchOp(df, "id string, c0 string, c1 string");
VectorAssembler res = new VectorAssembler()
.setSelectedCols("c0", "c1")
.setOutputCol("table2vec");
res.transform(data).print();输出结果如下所示：
id | c0 | c1 | table2vec |
|---|---|---|---|
0 | $6$1:2.0 2:3.0 5:4.3 | 3.0 2.0 3.0 | $9$1:2.0 2:3.0 ... 8:3.0 |
1 | $8$1:2.0 2:3.0 7:4.3 | 3.0 2.0 3.0 | $11$1:2.0 2:3.0 ... 10:3.0 |
2 | $8$1:2.0 2:3.0 7:4.3 | 2.0 3.0 2.2 | $11$1:2.0 2:3.0 ... 10:2.2 |
特征哈希
FeatureHasher 是用于将多列特征进行哈希映射，将特征合并成特征向量的类。
对于每个样本，将选定的特征列的值进行哈希映射，得到特征向量的维度。每个特征列的值经过哈希函数映射到特征向量的对应维度，形成一个特征向量。
这种方式可以降低特征维度，减少内存和计算资源的消耗，尤其适用于高维稀疏特征。
不限于数值类型的数据
示例：
List<Row> df1 = Arrays.asList(
Row.of(1.1, true, 2, "A"),
Row.of(1.1, false, 2, "B"),
Row.of(1.1, true, 1, "B"),
Row.of(2.2, true, 1, "A")
);
BatchOperator <?> inOp = new MemSourceBatchOp(df1, "double double, bool boolean, number int, str string");
FeatureHasher binarizer = new FeatureHasher().setSelectedCols("double", "bool", "number", "str").setOutputCol("output").setNumFeatures(3);输出结果如下所示：
double | bool | number | str | output |
|---|---|---|---|---|
1.1000 | true | 2 | A | $3$0:3.0 1:1.1 2:1.0 |
1.1000 | false | 2 | B | $3$0:3.0 1:1.1 2:1.0 |
1.1000 | true | 1 | B | $3$0:2.0 1:1.1 2:1.0 |
2.2000 | true | 1 | A | $3$0:2.0 1:2.2 2:1.0 |
文本特征生成
DocCountVectorizer 用于将文本数据转换为文本特征向量。它主要适用于文本数据的特征抽取和处理，以便进一步用于机器学习模型的训练和预测。主要用于自然语言处理（NLP）任务，如文本分类、情感分析、主题建模等。
特征抽取过程：
- 对于每个文本数据，
DocCountVectorizer统计每个词汇在文本中的出现次数，作为特征向量的值。 - 每个文本对应一个特征向量，特征向量的维度为词汇表的大小。
Alink库封装的DocCountVectorizer支持生成多种类型的特征向量，支持IDF/WORD_COUNT/TF_IDF/Binary/TF。默认是WORD_COUNT特征类型，即单词出现的频率。
示例：
List <Row> df = Arrays.asList(
Row.of(0, "二手旧书:医学电磁成像"),
Row.of(1, "二手美国文学选读（ 下册 ）李宜燮南开大学出版社 9787310003969"),
Row.of(2, "二手正版图解象棋入门/谢恩思主编/华龄出版社"),
Row.of(3, "二手中国糖尿病文献索引"),
Row.of(4, "二手郁达夫文集（ 国内版 ）全十二册馆藏书")
);
BatchOperator <?> inOp = new MemSourceBatchOp(df, "id int, text string");
Pipeline pipeline = new Pipeline()
.add(new Segment().setSelectedCol("text")) // 分词
.add(new DocCountVectorizer().setFeatureType("TF_IDF").setSelectedCol("text").setOutputCol("vec"));
pipeline.fit(inOp).transform(inOp).print();输出结果如下所示：
id | text | vec |
|---|---|---|
0 | 二手 旧书 : 医学 电磁 成像 | $37$10:0.1831020481113516 14:0.1831020481113516 ... 34:0.1831020481113516 |
1 | 二手 美国 文学 选读 （ 下册 ） 李宜燮 南开大学 出版社 9787310003969 | $37$0:0.04077336356234972 1:0.04077336356234972 ... 35:0.06462425227459469 |
2 | 二手 正版 图解 象棋 入门 / 谢恩 思 主编 / 华龄 出版社 | $37$5:0.0915510240556758 6:0.0915510240556758 ... 36:0.1831020481113516 |
3 | 二手 中国 糖尿病 文献 索引 | $37$8:0.21972245773362198 9:0.21972245773362198 ... 32:0.21972245773362198 |
4 | 二手 郁达夫 文集 （ 国内 版 ） 全 十二册 馆藏 书 | $37$0:0.046209812037329684 1:0.046209812037329684 ... 30:0.07324081924454065 |
文本哈希特征生成
DocHashCountVectorizer 将文本数据中的每个词汇通过哈希函数映射到特征向量的维度，得到特征向量。
特征向量的维度由用户设置的参数 numFeatures 决定，通常较小以减少维度。
示例：
List <Row> df = Arrays.asList(
Row.of(0, "二手旧书:医学电磁成像"),
Row.of(1, "二手美国文学选读（ 下册 ）李宜燮南开大学出版社 9787310003969"),
Row.of(2, "二手正版图解象棋入门/谢恩思主编/华龄出版社"),
Row.of(3, "二手中国糖尿病文献索引"),
Row.of(4, "二手郁达夫文集（ 国内版 ）全十二册馆藏书")
);
BatchOperator <?> inOp = new MemSourceBatchOp(df, "id int, text string");
Pipeline pipeline = new Pipeline()
.add(new Segment().setSelectedCol("text"))
.add(new DocHashCountVectorizer().setFeatureType("TF_IDF").setSelectedCol("text").setOutputCol("vec").setNumFeatures(3));
pipeline.fit(inOp).transform(inOp).print();输出结果如下所示：
id | text | vec |
|---|---|---|
0 | 二手 旧书 : 医学 电磁 成像 | $3$0:-0.7331685343967134 1:-0.5207269374140805 |
1 | 二手 美国 文学 选读 （ 下册 ） 李宜燮 南开大学 出版社 9787310003969 | $3$0:-0.776296095243579 1:-0.24504797054780258 2:-0.21559781926450705 |
2 | 二手 正版 图解 象棋 入门 / 谢恩 思 主编 / 华龄 出版社 | $3$0:-0.7331685343967134 1:-0.17357564580469348 2:-0.30543024395805163 |
3 | 二手 中国 糖尿病 文献 索引 | $3$0:-0.2932674137586854 1:-0.4165815499312644 2:-0.366516292749662 |
4 | 二手 郁达夫 文集 （ 国内 版 ） 全 十二册 馆藏 书 | $3$0:-0.5865348275173707 1:-0.34715129160938696 2:-0.24434419516644132 |
Word2Vec
Word2Vec是Google在2013年开源的一个将词表转为向量的算法，其利用神经网络，可以通过训练，将词映射到K维度空间向量，它主要用于将单词表示成高维空间中的向量，以便能够在计算机上更好地处理自然语言文本。Word2Vec 提供了一种有效的方式来捕获单词之间的语义关系，这对于自然语言处理任务非常有用。
Word2Vec有两种主要模型：Skip-gram和Continuous Bag of Words (CBOW)。 Skip-gram模型是预测上下文单词，而CBOW模型是预测中心单词。
示例：
List <Row> df = Arrays.asList(
Row.of("A B C")
);
BatchOperator <?> inOp = new MemSourceBatchOp(df, "tokens string");
Word2Vec word2vec = new Word2Vec().setSelectedCol("tokens").setMinCount(1).setVectorSize(2).setOutputCol("vec");
word2vec.fit(inOp).transform(inOp).print();输出结果如下所示：
tokens | vec |
|---|---|
A B C | 0.7309789158041142 0.40841706187852966 |