diff --git a/PyPI/setup.cfg b/PyPI/setup.cfg index 1d779f0..0b2aa44 100644 --- a/PyPI/setup.cfg +++ b/PyPI/setup.cfg @@ -1,7 +1,7 @@ [metadata] # replace with your username: name = guan -version = 0.1.92 +version = 0.1.93 author = guanjihuan author_email = guanjihuan@163.com description = An open source python package diff --git a/PyPI/src/guan.egg-info/PKG-INFO b/PyPI/src/guan.egg-info/PKG-INFO index c90c1d3..d0f8b8b 100644 --- a/PyPI/src/guan.egg-info/PKG-INFO +++ b/PyPI/src/guan.egg-info/PKG-INFO @@ -1,6 +1,6 @@ Metadata-Version: 2.1 Name: guan -Version: 0.1.92 +Version: 0.1.93 Summary: An open source python package Home-page: https://py.guanjihuan.com Author: guanjihuan diff --git a/PyPI/src/guan/machine_learning.py b/PyPI/src/guan/machine_learning.py index 96c94d3..47b8a61 100644 --- a/PyPI/src/guan/machine_learning.py +++ b/PyPI/src/guan/machine_learning.py @@ -191,4 +191,21 @@ def load_train_data(x_train, y_train, batch_size=32): from torch.utils.data import DataLoader, TensorDataset train_dataset = TensorDataset(x_train, y_train) train_loader = DataLoader(train_dataset, batch_size=batch_size, shuffle=True) - return train_loader \ No newline at end of file + return train_loader + +# 数据的主成分分析PCA +def pca_of_data(data, n_components=None, standard=1): + from sklearn.decomposition import PCA + if standard==1: + from sklearn.preprocessing import StandardScaler + scaler = StandardScaler() + data_scaled = scaler.fit_transform(data) + else: + data_scaled = data + if n_components==None: + pca = PCA() + else: + pca = PCA(n_components=n_components) + data_transformed = pca.fit_transform(data_scaled) + explained_variance_ratio = pca.explained_variance_ratio_ + return data_transformed, explained_variance_ratio \ No newline at end of file