Advertisement
Guest User

Untitled

a guest
Mar 29th, 2017
59
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.17 KB | None | 0 0
  1. import pandas as pd
  2. import numpy as np
  3. import matplotlib.pyplot as plt
  4. from sklearn import preprocessing
  5. from sklearn.naive_bayes import GaussianNB
  6. from sklearn.svm import SVC
  7. from sklearn import svm
  8. from sklearn.metrics import accuracy_score
  9. from sklearn.model_selection import train_test_split
  10. from sklearn.model_selection import GridSearchCV
  11. from sklearn.model_selection import StratifiedShuffleSplit
  12.  
  13. data=pd.read_csv("./kaggle/data.csv")
  14. data=data.drop('Unnamed: 32',axis=1)
  15. data=data.fillna(data.mean())
  16.  
  17. X=np.array(data.drop('diagnosis',axis=1))
  18. Y=np.array(data['diagnosis'])
  19. X_train, X_test, y_train, y_test = train_test_split(X, Y, test_size=0.25, random_state=42)
  20.  
  21.  
  22. def svc_param_selection(X, y, nfolds,X_test,y_test):
  23. C_range = np.logspace(-2, 10, 13)
  24. gamma_range = np.logspace(-9, 3, 13)
  25. param_grid = dict(gamma=gamma_range, C=C_range)
  26.  
  27. grid_search = GridSearchCV(svm.SVC(), param_grid, cv=nfolds)
  28. grid_search.fit(X, y)
  29. predicted=grid_search.predict(X_test)
  30. score=accuracy_score(y_test, predicted)
  31. return grid_search.best_params_
  32.  
  33.  
  34. cv = StratifiedShuffleSplit(n_splits=5, test_size=0.2, random_state=42)
  35. param=svc_param_selection(X_train,y_train,cv,X_test,y_test)
  36. print param
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement