Ejemplo n.º 1
0
def runTestsAndPrintReport(trainingCategories, testingCategories):
    
    allPositiveTrainingReviews = list();
    allNegativeTrainingReviews = list();
    allPositiveTestingReviews = list();
    allNegativeTestingReviews = list();
    
    for category in trainingCategories:
        (allPositiveTrainingReviewsTmp, allNegativeTrainingReviewsTmp) = getAllClassifiedReviews(AmazonReviewCorpusReader(), category)
        allPositiveTrainingReviews += allPositiveTrainingReviewsTmp;
        allNegativeTrainingReviews += allNegativeTrainingReviewsTmp;
    
    for category in testingCategories:
        (allPositiveTestingReviewsTmp, allNegativeTestingReviewsTmp) = getAllClassifiedReviews(AmazonReviewCorpusReader(), category)
        allPositiveTestingReviews += allPositiveTestingReviewsTmp;
        allNegativeTestingReviews += allNegativeTestingReviewsTmp;
    
    datasets = [ClassifiedDataset(0.1, 'POS', allPositiveTrainingReviews, allPositiveTestingReviews),
            ClassifiedDataset(0.1, 'NEG', allNegativeTrainingReviews, allNegativeTestingReviews)
            ]
    
    print 'training: %s vs testing: %s' % (trainingCategories, testingCategories);

    report = Report(datasets, factories)
    
    reportStats = report.getReport();
    
    Report.printReport(reportStats);
Ejemplo n.º 2
0
 def testGetReport(self):
     rep = Report()
     self.assertTrue(rep.saveReport("Escola Santa Clara","Juan","professor",0,1,0,1,0,2,1,0,0,1))
     form = {}
     form['qualification'] = 'professor'
     form['teacherName'] = 'Juan'
     form['question0'] = 0
     form['question1'] = 1
     form['question2'] = 0
     form['question3'] = 1
     form['question4'] = 0
     form['question5'] = 2
     form['question6'] = 1
     form['question7'] = 0
     form['question8'] = 0
     form['question9'] = 1
     self.assertEqual(rep.getReport('Escola Santa Clara'),form)