以下代码来自链接:Pytsk,本人只是对其进行学习。使用python的TSK工具进行模型搭建
import numpy as np #数学计算
import torch.nn as nn#损失函数或神经网络模型的搭建,构建深度学习和神经网络结合的模型
from sklearn.datasets import make_classification#机器学习分类数据集制造
from sklearn.metrics import accuracy_score#计算准确率
from sklearn.model_selection import train_test_split#数据集划分
from sklearn.preprocessing import StandardScaler#数据预处理,标准化
from torch.optim import AdamW#优化器
'''作者自己写的TSK库,主要是计算先验参数以及模型训练'''
from pytsk.gradient_descent.antecedent import AntecedentGMF, antecedent_init_center#模糊集构建和中心值和标准差的计算
from pytsk.gradient_descent.callbacks import EarlyStoppingACC#停止条件
from pytsk.gradient_descent.training import Wrapper#模型训练
from pytsk.gradient_descent.tsk import TSK#基本TSK模糊系统
#准备数据集训练
X, y