def test_dame_namsor_accuracy_score_dame(self): dn = DameNamsor() score1 = dn.accuracy_score_dame([1, 1], [1, 1]) self.assertEqual(score1, 1) score2 = dn.accuracy_score_dame([1, 1, 1, 0], [1, 1, 2, 0]) self.assertEqual(score2, 0.75) score3 = dn.accuracy_score_dame([1, 1, 1, 1, 2, 1], [1, 1, 1, 1, 2, 1]) self.assertEqual(score3, 1)
def test_dame_namsor_accuracy_score_dame_method_returns_correct_result( self): g = DameNamsor() score1 = g.accuracy_score_dame([1, 1], [1, 1]) self.assertEqual(score1, 1) score2 = g.accuracy_score_dame([1, 1, 1, 0], [1, 1, 2, 0]) self.assertEqual(score2, 0.75) score3 = g.accuracy_score_dame([1, 1, 1, 1, 2, 1], [1, 1, 1, 1, 2, 1]) self.assertEqual(score3, 1)
parser = argparse.ArgumentParser() parser.add_argument('--csv', default="files/names/min.csv") parser.add_argument('--api', default="damegender", choices=['customsearch', 'namsor', 'genderize', 'genderguesser', 'damegender', 'genderapi', 'nameapi']) parser.add_argument('--ml', default="nltk", choices=['nltk', 'svc', 'sgd', 'gaussianNB', 'multinomialNB', 'bernoulliNB']) args = parser.parse_args() if (args.api == "all"): dn = DameNamsor() print("################### Namsor!!") gl = dn.gender_list(path=args.csv) print("Gender list: " + str(gl)) sl = dn.guess_list(path=args.csv, binary=True) print("Guess list: " +str(sl)) namsor_accuracy = dn.accuracy_score_dame(gl, sl) #print(dn.accuracy_score_dame([1, 1, 1, 1, 2, 1], [1, 1, 1, 1, 1, 1])) #namsor_accuracy = dn.accuracy_score_dame([1, 1, 1, 1, 2, 1], [1, 1, 1, 1, 2, 1]) print("Namsor accuracy: %s" % namsor_accuracy) dg = DameGenderize() print("################### Genderize!!") gl = dg.gender_list(path=args.csv) print("Gender list: " + str(gl)) sl = dg.guess_list(path=args.csv, binary=True) print("Guess list: " +str(gl)) genderize_accuracy = dg.accuracy_score_dame(gl,sl) print("Genderize accuracy: %s" % genderize_accuracy)