import fpGrowth rootNode = fpGrowth.treeNode('pyramid', 9, None) rootNode.children['eye'] = fpGrowth.treeNode('eye', 13, None) rootNode.disp() rootNode.children['phoenix'] = fpGrowth.treeNode('phoenix', 3, None) rootNode.disp() from importlib import reload reload(fpGrowth) simpDat = fpGrowth.loadSimpDat() simpDat initSet = fpGrowth.createInitSet(simpDat) initSet # 创建FP树 myFPtree, myHeaderTab = fpGrowth.createTree(initSet, 3) myFPtree.disp() reload(fpGrowth) fpGrowth.findPrefixPath('x', myHeaderTab['x'][1]) fpGrowth.findPrefixPath('z', myHeaderTab['z'][1]) fpGrowth.findPrefixPath('r', myHeaderTab['r'][1]) reload(fpGrowth) freqItems = [] fpGrowth.mineTree(myFPtree, myHeaderTab, 3, set([]), freqItems) freqItems # 示例:从新闻网站点击流中挖掘 parsedDat = [line.split() for line in open('kosarak.dat').readlines()] initSet = fpGrowth.createInitSet(parsedDat) myFPtree, myHeaderTab = fpGrowth.createTree(initSet, 100000)
#!/usr/bin/env python3 # -*- coding: UTF-8 -*- import fpGrowth rootNode = fpGrowth.treeNode("pyramid", 9, None) # 这个调用的就是class的__init__函数来声明第一个结点 rootNode.children["eye"] = fpGrowth.treeNode("eye", 13, None) # rootNode的['eye']孩子结点也是新声明的一个treeNode rootNode.display() rootNode.children["phoenix"] = fpGrowth.treeNode("phoenix", 3, rootNode.children["eye"]) rootNode.display() simpDat = fpGrowth.loadSimpDat() print(simpDat) initSet = fpGrowth.createInitSet(simpDat) print("createTree with this initSet:", initSet) myFPtree, myHeaderTab = fpGrowth.createTree(initSet, 3) myFPtree.display() Result_x = fpGrowth.findPrefixPath("x", myHeaderTab["x"][1]) Result_z = fpGrowth.findPrefixPath("z", myHeaderTab["z"][1]) Result_r = fpGrowth.findPrefixPath("r", myHeaderTab["r"][1]) print(Result_x) print(Result_z) print(Result_r) freqItems = [] Result = fpGrowth.mineTree(myFPtree, myHeaderTab, 3, set([]), freqItems) print(Result) print("100万记录用FP-growth来处理:\n") parsedDat = [line.split() for line in open("kosarak.dat").readlines()]
import fpGrowth rootNode = fpGrowth.treeNode('pyramid', 9, None) rootNode.children['eye'] = fpGrowth.treeNode('eye', 13, None) rootNode.disp() rootNode.children['phoenix'] = fpGrowth.treeNode('phoenix', 3, None) rootNode.disp() simpDat = fpGrowth.loadSimpDat() initSet = fpGrowth.createInitSet(simpDat) print initSet myFPtree, myHeaderTab = fpGrowth.createTree(initSet, 3) myFPtree.disp() print fpGrowth.findPrefixPath('x', myHeaderTab['x'][1]) print fpGrowth.findPrefixPath('z', myHeaderTab['z'][1]) print fpGrowth.findPrefixPath('r', myHeaderTab['r'][1]) freqItems = [] fpGrowth.mineTree(myFPtree, myHeaderTab, 3, set([]), freqItems) print freqItems
#coding=utf-8 import fpGrowth from numpy import * if __name__ == '__main__': rootNode=fpGrowth.treeNode('p',9,None) rootNode.children['eye']=fpGrowth.treeNode('eye',13,None) rootNode.disp()
def test(): rootNode = fpGrowth.treeNode('pyramid', 9, None) rootNode.children['eye'] = fpGrowth.treeNode('eye', 13, None) rootNode.children['phoeix'] = fpGrowth.treeNode('phoeix', 3, None) rootNode.disp()
class treeNode: def __init__(self,nameValue,numOccur,parentNode): self.name = nameValue self.count = numOccur self.nodeLink = None self.parent = parentNode self.children = {} def inc(self,numOccur): self.count += numOccur def disp(self,ind=1): print(' '*ind,self.name,' ',self.count) for child in self.children.values(): child.disp(ind+1) def updateTree(items, inTree, headerTable, count): if items[0] in inTree.children: # check if orderedItems[0] in retTree.children inTree.children[items[0]].inc(count) # incrament count else: # add items[0] to inTree.children inTree.children[items[0]] = treeNode(items[0], count, inTree) if headerTable[items[0]][1] == None: # update header table headerTable[items[0]][1] = inTree.children[items[0]] else: updateHeader(headerTable[items[0]][1], inTree.children[items[0]]) if len(items) > 1: # call updateTree() with remaining ordered items updateTree(items[1::], inTree.children[items[0]], headerTable, count) def updateHeader(nodeToTest, targetNode): # this version does not use recursion while (nodeToTest.nodeLink != None): # Do not use recursion to traverse a linked list! nodeToTest = nodeToTest.nodeLink nodeToTest.nodeLink = targetNode def ascendTree(leafNode, prefixPath): # ascends from leaf node to root if leafNode.parent != None: prefixPath.append(leafNode.name) ascendTree(leafNode.parent, prefixPath) def findPrefixPath(basePat, treeNode): # treeNode comes from header table condPats = {} while treeNode != None: prefixPath = [] ascendTree(treeNode, prefixPath) if len(prefixPath) > 1: condPats[frozenset(prefixPath[1:])] = treeNode.count treeNode = treeNode.nodeLink return condPats def mineTree(inTree, headerTable, minSup, preFix, freqItemList): bigL = [v[0] for v in sorted(headerTable.items(), key=lambda p: p[1])] # (sort header table) for basePat in bigL: # start from bottom of header table newFreqSet = preFix.copy() newFreqSet.add(basePat) # print 'finalFrequent Item: ',newFreqSet #append to set freqItemList.append(newFreqSet) condPattBases = findPrefixPath(basePat, headerTable[basePat][1]) # print 'condPattBases :',basePat, condPattBases # 2. construct cond FP-tree from cond. pattern base myCondTree, myHead = createTree(condPattBases, minSup) # print 'head from conditional tree: ', myHead if myHead != None: # 3. mine cond. FP-tree # print 'conditional tree for: ',newFreqSet # myCondTree.disp(1) mineTree(myCondTree, myHead, minSup, newFreqSet, freqItemList) def loadSimpDat(): simpDat = [['r', 'z', 'h', 'j', 'p'], ['z', 'y', 'x', 'w', 'v', 'u', 't', 's'], ['z'], ['r', 'x', 'n', 'o', 's'], ['y', 'r', 'x', 'z', 'q', 't', 'p'], ['y', 'z', 'x', 'e', 'q', 's', 't', 'm']] return simpDat def createInitSet(dataSet): retDict = {} for trans in dataSet: retDict[frozenset(trans)] = 1 return retDict import twitter from time import sleep import re def textParse(bigString): urlsRemoved = re.sub('(http:[/][/]|www.)([a-z]|[A-Z]|[0-9]|[/.]|[~])*', '', bigString) listOfTokens = re.split(r'\W*', urlsRemoved) return [tok.lower() for tok in listOfTokens if len(tok) > 2] def getLotsOfTweets(searchStr): CONSUMER_KEY = '' CONSUMER_SECRET = '' ACCESS_TOKEN_KEY = '' ACCESS_TOKEN_SECRET = '' api = twitter.Api(consumer_key=CONSUMER_KEY, consumer_secret=CONSUMER_SECRET, access_token_key=ACCESS_TOKEN_KEY, access_token_secret=ACCESS_TOKEN_SECRET) # you can get 1500 results 15 pages * 100 per page resultsPages = [] for i in range(1, 15): print("fetching page %d" % i) searchResults = api.GetSearch(searchStr, per_page=100, page=i) resultsPages.append(searchResults) sleep(6) return resultsPages def mineTweets(tweetArr, minSup=5): parsedList = [] for i in range(14): for j in range(100): parsedList.append(textParse(tweetArr[i][j].text)) initSet = createInitSet(parsedList) myFPtree, myHeaderTab = createTree(initSet, minSup) myFreqList = [] mineTree(myFPtree, myHeaderTab, minSup, set([]), myFreqList) return myFreqList # minSup = 3 # simpDat = loadSimpDat() # initSet = createInitSet(simpDat) # myFPtree, myHeaderTab = createTree(initSet, minSup) # myFPtree.disp() # myFreqList = [] # mineTree(myFPtree, myHeaderTab, minSup, set([]), myFreqList) if __name__ == "__main__": import fpGrowth rootNode = fpGrowth.treeNode('pyramid',9,None) rootNode.children['eye']=fpGrowth.treeNode('eye',13,None) rootNode.disp()