def test_dame_namsor_guess_list_method_returns_correct_result(self):
g = DameNamsor()
self.assertEqual([
'male', 'male', 'male', 'male', 'male', 'male', 'female', 'female',
'male', 'male', 'male', 'male', 'male', 'male', 'male', 'male',
'male', 'male', 'female', 'male', 'male'
], g.guess_list(path="files/names/partial.csv", binary=False))
self.assertEqual(
[1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1],
g.guess_list(path="files/names/partial.csv", binary=True))
def test_dame_namsor_guess_list(self):
dn = DameNamsor()
if (dn.config['DEFAULT']['namsor'] == 'yes'):
self.assertEqual([
'male', 'male', 'male', 'male', 'male', 'male', 'female',
'female', 'male', 'male', 'male', 'male', 'male', 'male',
'male', 'male', 'male', 'male', 'female', 'male', 'male'
], dn.guess_list(path="files/names/partial.csv", binary=False))
self.assertEqual([
1, 1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 1, 1
], dn.guess_list(path="files/names/partial.csv", binary=True))
from app.dame_customsearch import DameCustomsearch
import argparse
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))
