File tree 1 file changed +40
-0
lines changed
1 file changed +40
-0
lines changed Original file line number Diff line number Diff line change
1
+ # -*- coding: utf-8 -*-
2
+ """
3
+ Created on Sun Aug 12 14:49:47 2018
4
+
5
+ @author: Administrator
6
+ """
7
+
8
+
9
+ from sklearn .model_selection import train_test_split
10
+ from sklearn .preprocessing import MinMaxScaler , StandardScaler
11
+ from sklearn .decomposition import PCA
12
+ from sklearn .svm import LinearSVC , SVC
13
+ from sklearn .datasets import load_breast_cancer
14
+
15
+ def test_PCA ():
16
+
17
+ cancer = load_breast_cancer ()
18
+
19
+ X_train , X_test , y_train , y_test = train_test_split (
20
+ cancer .data , cancer .target , stratify = cancer .target
21
+ , random_state = 42 )
22
+
23
+ svc = SVC (C = 100 )
24
+ svc .fit (X_train ,y_train )
25
+
26
+ print (svc .score (X_train ,y_train ))
27
+ print ("the test score is {:.2f}" .format (svc .score (X_test
28
+ ,y_test )))
29
+
30
+ X_tmp = StandardScaler ().fit_transform (X_train )
31
+ scale = PCA (n_components = 2 )
32
+ scale .fit (X_tmp )
33
+ X_test_scaled = scale .transform (X_test )
34
+ X_scaled = scale .transform (X_train )
35
+
36
+ svc .fit (X_scaled , y_train )
37
+ print ("the test score scaled is {:.2f}" .format (svc .score (X_test_scaled
38
+ ,y_test )))
39
+
40
+
You can’t perform that action at this time.
0 commit comments