apriori_pattern_miner 0.1.1

Implementation of Apriori Pattern Mining algorithm
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

// THIS CODE IS MY OWN WORK, IT WAS WRITTEN
// WITHOUT CONSULTING CODE WRITTEN BY OTHER STUDENTS ~ Mikolaj Figurski

extern crate csv;

use std::collections::HashMap;
use std::io;

pub type KeywordSupport = HashMap<String, Vec<u32>>;

pub fn read_keyword_support(
    filename: &str,
    min_support: usize,
) -> Result<KeywordSupport, io::Error> {
    #[derive(Deserialize)]
    struct Record {
        text_keywords: String,
    }

    let mut rdr = csv::Reader::from_path(filename)?;
    let mut base = KeywordSupport::new();
    let mut i: u32 = 0;
    for result in rdr.deserialize() {
        let record: Record = result?;
        for keyword in record.text_keywords.split(';') {
            // insert keyword into base
            base.entry(keyword.to_string()).or_insert(vec![]).push(i);
        }
        i += 1;
    }
    // clear items below threshold support
    base.retain(|_, v| v.len() >= min_support);
    Ok(base)
}