from sandbox.util.Util import Util
from sandbox.util.Evaluator import Evaluator
#logging.basicConfig(stream=sys.stdout, level=logging.DEBUG)
ranLSI = True
printOutputLists = False
printPrecisions = False
printDocuments = True
numpy.set_printoptions(suppress=True, precision=3, linewidth=100)
dataset = ArnetMinerDataset(runLSI=ranLSI)
#dataset.fields = ["Intelligent Agents"]
if printDocuments:
print("Reading article data")
authorList, documentList, citationList = dataset.readAuthorsAndDocuments(useAbstract=False)
print("Done")
ns = numpy.arange(5, 55, 5)
bestaverageTestPrecisions = numpy.zeros(len(dataset.fields))
computeInfluence = True
graphRanker = GraphRanker(k=100, numRuns=100, computeInfluence=computeInfluence, p=0.05, inputRanking=[1, 2])
methodNames = graphRanker.getNames()
methodNames.append("MC2")
numMethods = len(methodNames)
averageTrainPrecisions = numpy.zeros((len(dataset.fields), len(ns), numMethods))
averageTestPrecisions = numpy.zeros((len(dataset.fields), len(ns), numMethods))
coverages = numpy.load(dataset.coverageFilename)
