47 from daal.algorithms
import association_rules
48 from daal.data_management
import FileDataSource, DataSourceIface
50 utils_folder = os.path.realpath(os.path.abspath(os.path.dirname(os.path.dirname(__file__))))
51 if utils_folder
not in sys.path:
52 sys.path.insert(0, utils_folder)
53 from utils
import printAprioriItemsets, printAprioriRules
56 datasetFileName = os.path.join(
'..',
'data',
'batch',
'apriori.csv')
63 dataSource = FileDataSource(
64 datasetFileName, DataSourceIface.doAllocateNumericTable, DataSourceIface.doDictionaryFromContext
68 dataSource.loadDataBlock()
71 alg = association_rules.Batch()
72 alg.input.set(association_rules.data, dataSource.getNumericTable())
73 alg.parameter.minSupport = minSupport
74 alg.parameter.minConfidence = minConfidence
80 nt1 = res.get(association_rules.largeItemsets)
81 nt2 = res.get(association_rules.largeItemsetsSupport)
83 nt3 = res.get(association_rules.antecedentItemsets)
84 nt4 = res.get(association_rules.consequentItemsets)
85 nt5 = res.get(association_rules.confidence)
87 printAprioriItemsets(nt1, nt2)
88 printAprioriRules(nt3, nt4, nt5)