#Find all of the decays in information transmission egoTestFileName = dataDir + "EgoInfoTest.csv" alterTestFileName = dataDir + "AltersInfoTest.csv" decayGraph = eCsvReader.findInfoDecayGraph(egoTestFileName, alterTestFileName, egoIndicesR, alterIndices, egoIndicesNR, alterIndicesNR, egoFileName, alterFileName, missing) logging.info("Size of decay graph: " + str(decayGraph.getNumVertices())) logging.info("Number of edges: " + str(decayGraph.getNumEdges())) #Now write the decays to a simplegraph file decayFileName = dataDir + "EgoAlterDecays.dat" decayGraph.save(decayFileName) logging.info("Wrote decays to file " + decayFileName) #Now write out transmissions sampleSize = 1000 indices = Util.sampleWithoutReplacement(sampleSize, examplesList.getNumExamples()) examplesListSample = examplesList.getSubExamplesList(indices) outputFileName1 = dataDir + "EgoAlterTransmissions" outputFileName2 = dataDir + "EgoAlterTransmissions1000" examplesList.writeToMatFile(outputFileName1) examplesListSample.writeToMatFile(outputFileName2) #Let's also write out the csv file for analysis X = examplesList.getDataField(examplesList.getDefaultExamplesName()) y = examplesList.getDataField(examplesList.getLabelsName()) X = numpy.c_[X, y]