Skip to content

Instantly share code, notes, and snippets.

@szilard

szilard/mlbenchm-py-RF-int-enc.py

Created May 6, 2015 21:01
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save szilard/56706595b4594e297414 to your computer and use it in GitHub Desktop.
Save szilard/56706595b4594e297414 to your computer and use it in GitHub Desktop.
Download ZIP
Scikit-learn RF with integer encoded categoricals
Raw
mlbenchm-py-RF-int-enc.py
import numpy as np
import pandas as pd
from sklearn.ensemble import RandomForestClassifier
from sklearn import metrics
d_train = pd.read_csv("train-intcateg-1m.csv", header=None)
d_test = pd.read_csv("test-intcateg-1m.csv", header=None)
X_train = d_train.ix[:,0:7]
y_train = d_train.ix[:,8]
X_test = d_test.ix[:,0:7]
y_test = d_test.ix[:,8]
md = RandomForestClassifier(n_estimators = 500, n_jobs = -1)
%time md.fit(X_train, y_train)
%time phat = md.predict_proba(X_test)[:,1]
metrics.roc_auc_score(y_test, phat)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment