bagos-apriori/Apriori.js

Apriori adalah suatu metode Data Mining

// Apriori.js

var _testDB = [
    // 'cheese, diaper, water, bread, umbrella',
    // 'diaper, water',
    // 'cheese, diaper, milk',
    // 'diaper, cheese, detergent',
    // 'cheese, milk, beer'
    'beer, detergent, diaper',
    'beer, cheese, umbrella, bread, water, diaper, milk',
    'milk, umbrella',
    'beer, umbrella, water',
    'bread, cheese, water',
    'cheese, bread, water, diaper, umbrella',
    'water, diaper, beer',
    'bread, beer, umbrella, cheese, water, diaper, milk',
    'water, milk, umbrella, bread',
    'diaper, umbrella',
];

const apriori = function(){
    //Create ItemsetCollection for current db
    let db = new ItemsetCollection();
    for (var i in _testDB) {
        let items = _testDB[i].split(', ');
        db.push(Itemset.from(items));
    }


    // Step1: Find large itemsets for given support threshold
    let supportThreshold = parseFloat(40);
    let L = AprioriMining.doApriori(db, supportThreshold);


    // Step2: Build rules based on large itemsets and confidence threshold
    let confidenceThreshold = parseFloat(70);
    let allRules = AprioriMining.mine(db, L, confidenceThreshold);

    console.log(L)
    console.log(allRules);
}

module.exports =  apriori;