Skip to content

Commit

Permalink
Random forest classifiers
Browse files Browse the repository at this point in the history
  • Loading branch information
shahanesanket committed Apr 14, 2017
1 parent 3973323 commit 9bfb645
Showing 1 changed file with 12 additions and 1 deletion.
13 changes: 12 additions & 1 deletion Python/rough-sanket.py
Original file line number Diff line number Diff line change
Expand Up @@ -78,4 +78,15 @@ def plotCorrMatrix(data,title='Continuous Feature Correlation Figure 1'):

original_pca = pd.read_csv('data/pca_data.csv')

exp_variance = list(pca.explained_variance_ratio_)
exp_variance = list(pca.explained_variance_ratio_)

# Try random forest and other classification techniques like SVM
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import cross_val_score
import pandas as pd
import numpy as np

training_data = pd.read_csv('../data/pca_data.csv')
rf = RandomForestClassifier(n_estimators = 10, n_jobs=-1)
scores = cross_val_score(rf, training_data.iloc[:,:-1], training_data.iloc[:,-1], cv = 10)
print scores

0 comments on commit 9bfb645

Please sign in to comment.