Guest User

Untitled

a guest
Aug 17th, 2018
111
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.47 KB | None | 0 0
  1. import pandas as pd
  2. from sklearn import svm
  3. from sklearn.model_selection import train_test_split
  4.  
  5. df = pd.DataFrame({'age':[20,30,40,50],
  6. 'sex':['male','female','female','male'],
  7. 'region':['northwest','southwest','northeast','southeast'],
  8. 'charges':[1000,1000,2000,2000]})
  9. df.sex = (df.sex == 'female')
  10. df.region = pd.Categorical(df.region)
  11. df.region = df.region.cat.codes
  12. X = df.loc[:,['age','sex','region']]
  13. y = df.loc[:,['charges']]
  14.  
  15. X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33, random_state=42)
  16.  
  17. clf = svm.SVC(C=1.0, cache_size=200,decision_function_shape='ovr', degree=3, gamma='auto', kernel='rbf')
  18. clf.fit(X_train, y_train)
  19.  
  20. import pandas as pd
  21. from sklearn import svm
  22. from sklearn.model_selection import train_test_split
  23.  
  24. df = pd.DataFrame({'age':[20,30,40,50],
  25. 'sex':['male','female','female','male'],
  26. 'region':['northwest','southwest','northeast','southeast'],
  27. 'charges':[1000,1000,2000,2000]})
  28. df.sex = (df.sex == 'female')
  29. df = pd.concat([df,pd.get_dummies(df.region)],axis = 1).drop('region',1)
  30. X = df.drop('charges',1)
  31. y = df.charges
  32. X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33, random_state=42)
  33.  
  34. clf = svm.SVC(C=1.0, cache_size=200,decision_function_shape='ovr', degree=3, gamma='auto', kernel='rbf')
  35. clf.fit(X_train, y_train)
  36.  
  37. from sklearn.preprocessing import LabelEncoder
  38. le = LabelEncoder()
  39. df.region = le.fit_transform(df.region)
Add Comment
Please, Sign In to add comment