Module rustlearn::feature_extraction::dict_vectorizer

DictVectorizer vectorizers a set of named features into a sparse array via one-hot encoding.

Examples

use rustlearn::prelude::*;
use rustlearn::feature_extraction::DictVectorizer;

let features = vec![vec!["feature_one", "feature_two"],
                    vec!["feature_two"]];

let mut vectorizer = DictVectorizer::new();

for (row_idx, row) in features.iter().enumerate() {
    for feature in row.iter() {
        vectorizer.partial_fit(row_idx, feature, 1.0);
    }
}

let X = vectorizer.transform();

assert!(X.rows() == 2 && X.cols() == 2);

Structs

DictVectorizer