24 from daal.algorithms
import association_rules
25 from daal.data_management
import FileDataSource, DataSourceIface
27 utils_folder = os.path.realpath(os.path.abspath(os.path.dirname(os.path.dirname(__file__))))
28 if utils_folder
not in sys.path:
29 sys.path.insert(0, utils_folder)
30 from utils
import printAprioriItemsets, printAprioriRules
33 datasetFileName = os.path.join(
'..',
'data',
'batch',
'apriori.csv')
40 dataSource = FileDataSource(
41 datasetFileName, DataSourceIface.doAllocateNumericTable, DataSourceIface.doDictionaryFromContext
45 dataSource.loadDataBlock()
48 alg = association_rules.Batch()
49 alg.input.set(association_rules.data, dataSource.getNumericTable())
50 alg.parameter.minSupport = minSupport
51 alg.parameter.minConfidence = minConfidence
57 nt1 = res.get(association_rules.largeItemsets)
58 nt2 = res.get(association_rules.largeItemsetsSupport)
60 nt3 = res.get(association_rules.antecedentItemsets)
61 nt4 = res.get(association_rules.consequentItemsets)
62 nt5 = res.get(association_rules.confidence)
64 printAprioriItemsets(nt1, nt2)
65 printAprioriRules(nt3, nt4, nt5)