22 from daal.algorithms
import association_rules
23 from daal.data_management
import FileDataSource, DataSourceIface
25 utils_folder = os.path.realpath(os.path.abspath(os.path.dirname(os.path.dirname(__file__))))
26 if utils_folder
not in sys.path:
27 sys.path.insert(0, utils_folder)
28 from utils
import printAprioriItemsets, printAprioriRules
31 datasetFileName = os.path.join(
'..',
'data',
'batch',
'apriori.csv')
38 dataSource = FileDataSource(
39 datasetFileName, DataSourceIface.doAllocateNumericTable, DataSourceIface.doDictionaryFromContext
43 dataSource.loadDataBlock()
46 alg = association_rules.Batch()
47 alg.input.set(association_rules.data, dataSource.getNumericTable())
48 alg.parameter.minSupport = minSupport
49 alg.parameter.minConfidence = minConfidence
55 nt1 = res.get(association_rules.largeItemsets)
56 nt2 = res.get(association_rules.largeItemsetsSupport)
58 nt3 = res.get(association_rules.antecedentItemsets)
59 nt4 = res.get(association_rules.consequentItemsets)
60 nt5 = res.get(association_rules.confidence)
62 printAprioriItemsets(nt1, nt2)
63 printAprioriRules(nt3, nt4, nt5)