Packages

ex_keyword_extraction

0.1.0
0.1.0

ex_keyword_extraction is an elixir wrapper for keyword_extraction crate - it includes the following algorithms: TF-IDF, Rake, TextRank

HexDocs Report package

Current section

Files

Jump to
Readme
1 Version
2 Dependencies
Dependants
Files
Activity
ex_keyword_extraction native exkeywordextractionnative src tf_idf.rs
Raw

native/exkeywordextractionnative/src/tf_idf.rs 

use keyword_extraction::tf_idf::{TfIdf, TfIdfParams};

use std::collections::HashMap;



#[rustler::nif(schedule = "DirtyCpu")]

pub fn tfidf_get_word_scores_map(documents: Vec<String>) -> HashMap<String, f32> {

    let params = TfIdfParams::ProcessedDocuments(documents.as_slice());

    TfIdf::new(params).get_word_scores_map().to_owned()

}