Gender detection testing app (genderPredictor)

gender_detection.py

import csv
from genderPredictor import genderPredictor

gp = genderPredictor()
gp.trainAndTest()

def gender(name):
    output = 'unknown'

    tmp = gp.classify(name)
    if tmp == 'F':
        output = 'female'
    elif  tmp == 'M':
        output = 'male'
    else:
        output = 'unknown'
    return output


rows = []
with open('output.csv', 'r') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        rows.append(row)

with open('output.csv', 'w') as csvfile:
    writer = csv.writer(csvfile)
    for row in rows:
        row.append(gender(row[0]))
        writer.writerow(row)

Nice work...by the way from where i can get the separated male and female dataset...!

Nice work...by the way from where i can get the separated male and female dataset...!