pytorch 搭建的基於LSTM自編碼器對數據降維並採用KNN算法對鳶尾花分類

LSTM搭建自編碼器提取特徵,KNN分類 import torch import torch.nn as nn from sklearn.datasets import load_iris from sklearn.model_selection import train_test_split # 超參數 EPOCH = 200 LR = 0.005 data = load_iris() y
相關文章
相關標籤/搜索