In [66]:
import pandas as pd
import numpy as np
from sklearn.linear_model import LogisticRegression
from sklearn.linear_model import SGDClassifier
from sklearn.ensemble import RandomForestClassifier
from sklearn.preprocessing import LabelEncoder
from sklearn.preprocessing import OneHotEncoder
from sklearn import cross_validation
from sklearn.metrics import roc_auc_score
In [67]:
#donations = pd.read_csv('../data/donations.csv').sort('projectid')
projects = pd.read_csv('../data/projects.csv').sort('projectid')
outcomes = pd.read_csv('../data/outcomes.csv').sort('projectid')
#resources = pd.read_csv('../data/resources.csv').sort('projectid')
sample = pd.read_csv('../data/sampleSubmission.csv').sort('projectid')
#essays = pd.read_csv('../data/essays.csv').sort('projectid')
In [68]:
projects = projects.merge(outcomes, how ='inner')
dates = np.array(projects.date_posted)
train_idx = np.where((dates < '2013-05-01') & (dates >= '2010-01-01'))[0]
test_idx = np.where(dates >= '2013-05-01')[0]
outcomes = np.array(projects.is_exciting)
In [69]:
projects.secondary_focus_area = projects.secondary_focus_area.fillna(projects.primary_focus_area)
projects.secondary_focus_subject = projects.secondary_focus_subject.fillna(projects.primary_focus_subject)
projects = projects.fillna(method='pad')
projects['month'] = ''
for i in range(0,projects.shape[0]):
projects['month'][i] = projects.date_posted[i][5:7]
In [70]:
cols = ['school_city', 'school_state', 'school_zip', 'school_metro', 'school_district', 'school_county', 'school_charter', 'school_magnet',
'school_year_round', 'school_nlns', 'school_kipp', 'school_charter_ready_promise', 'teacher_prefix', 'teacher_teach_for_america', 'teacher_ny_teaching_fellow', 'primary_focus_subject','primary_focus_area',
'secondary_focus_subject', 'secondary_focus_area', 'resource_type', 'poverty_level', 'grade_level',
'students_reached', 'eligible_double_your_impact_match', 'eligible_almost_home_match','month' ]
projects = np.array(projects[cols])
In [71]:
for i in range(0, projects.shape[1]):
le = LabelEncoder()
projects[:,i] = le.fit_transform(projects[:,i])
projects = projects.astype(float)
In [72]:
train, crossval, outcomeTrain, outcomeCrossVal = projects[train_idx], projects[test_idx], outcomes[train_idx], outcomes[test_idx]
model = RandomForestClassifier(n_estimators = 100, criterion = 'entropy', n_jobs = -1)
model.fit(train, outcomeTrain=='t')
preds3 = model.predict_proba(crossval)[:,1]
In [73]:
ohe = OneHotEncoder()
projects = ohe.fit_transform(projects)
In [74]:
train, crossval, outcomeTrain, outcomeCrossVal = projects[train_idx], projects[test_idx], outcomes[train_idx], outcomes[test_idx]
In [75]:
model = LogisticRegression(C = .1)
model.fit(train, outcomeTrain=='t')
preds1 = model.predict_proba(crossval)[:,1]
model = SGDClassifier(alpha = 0.001, loss = 'modified_huber', penalty = 'l2', n_iter = 1000, n_jobs = -1)
model.fit(train, outcomeTrain=='t')
preds2 = model.predict_proba(crossval)[:,1]
In [76]:
preds = 0.6*preds1 + 0.2*preds2 + 0.2*preds3
In [77]:
value = roc_auc_score(outcomeCrossVal=='t', preds)
value
Out[77]: