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
相关文章
相关标签/搜索