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