Skip to content

Commit 892d510

Browse files
committed
PCA
1 parent 1a78693 commit 892d510

File tree

1 file changed

+40
-0
lines changed

1 file changed

+40
-0
lines changed

Diff for: Unsup_learn/PCA.py

+40
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,40 @@
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+

0 commit comments

Comments
 (0)