企业官网建设流程全解析

网站建设策划目的及过程,工程师招聘网站,论文网站建设方案,网站排名怎么弄首先#xff0c;将原始一维振动信号通过短时傅里叶变换转换为时频谱图#xff0c;形成二维图像特征#xff1b;接着构建了一个改进的UNet神经网络架构#xff0c;该网络在保留UNet编码器-解码器结构的基础上移除了时间嵌入模块#xff0c;增加了注意力机制和残差连接…首先将原始一维振动信号通过短时傅里叶变换转换为时频谱图形成二维图像特征接着构建了一个改进的UNet神经网络架构该网络在保留UNet编码器-解码器结构的基础上移除了时间嵌入模块增加了注意力机制和残差连接专门用于谱图特征提取和分类然后采用数据增强技术扩充训练样本通过分层抽样划分数据集模型训练阶段使用交叉熵损失函数、Adam优化器和学习率调度策略并在验证集上监控性能保存最佳模型最后在测试集上评估模型性能计算准确率、混淆矩阵等指标并可视化训练过程、预测结果和诊断报告实现对轴承正常、滚珠故障、内圈故障和外圈故障四种状态的准确分类诊断。import numpy as np import pandas as pd import matplotlib.pyplot as plt from pathlib import Path import warnings warnings.filterwarnings(ignore) import torch import torch.nn as nn from torch.utils.data import Dataset, DataLoader from torch.nn import functional as F import torch.optim as optim from torch.optim.lr_scheduler import ReduceLROnPlateau import math from sklearn.metrics import confusion_matrix, classification_report from sklearn.model_selection import train_test_split import seaborn as sns from scipy import signal import scipy.stats as stats # 设置随机种子确保可重复性 def set_seed(seed42): torch.manual_seed(seed) torch.cuda.manual_seed_all(seed) if torch.cuda.is_available() else None np.random.seed(seed) torch.backends.cudnn.deterministic True torch.backends.cudnn.benchmark False set_seed(42) # 定义设备 device torch.device(cuda if torch.cuda.is_available() else cpu) print(fUsing device: {device}) # 原始UNet组件 class Swish(nn.Module): Swish激活函数 def forward(self, x): return x * torch.sigmoid(x) class TimeEmbedding(nn.Module): 时间嵌入层在原始UNet中用于扩散模型这里保留但会简化 def __init__(self, T, d_model, dim): assert d_model % 2 0 super().__init__() emb torch.arange(0, d_model, step2) / d_model * math.log(10000) emb torch.exp(-emb) pos torch.arange(T).float() emb pos[:, None] * emb[None, :] assert list(emb.shape) [T, d_model // 2] emb torch.stack([torch.sin(emb), torch.cos(emb)], dim-1) assert list(emb.shape) [T, d_model // 2, 2] emb emb.view(T, d_model) self.timembedding nn.Sequential( nn.Embedding.from_pretrained(emb), nn.Linear(d_model, dim), Swish(), nn.Linear(dim, dim), ) self.initialize() def initialize(self): 初始化权重 for module in self.modules(): if isinstance(module, nn.Linear): nn.init.xavier_uniform_(module.weight) nn.init.zeros_(module.bias) def forward(self, t): emb self.timembedding(t) return emb详细算法步骤数据采集与预处理阶段从西储大学轴承数据集中加载四种不同故障状态的原始振动信号文件对每一类信号进行标准化处理消除量纲影响然后将长时信号按照固定长度和重叠率分割成多个短时信号片段为后续分析准备基础数据单元。特征工程构建阶段对每个信号片段应用短时傅里叶变换算法将一维时域信号转换为二维时频谱图通过对数变换增强特征对比度再统一缩放至固定尺寸形成标准化图像特征同时提取信号的时域统计特征和频域特征作为补充信息。数据集划分阶段采用分层抽样策略将特征数据集按比例划分为训练集、验证集和测试集确保每个子集中各类别样本分布均衡避免因数据划分不当导致的模型评估偏差。数据增强处理阶段在训练集上应用随机噪声添加和随机幅度缩放等数据增强技术人工扩充训练样本多样性提高模型对实际工况变化的适应能力和泛化性能。神经网络模型构建阶段设计改进型UNet分类网络架构保留编码器-解码器对称结构用于多尺度特征提取引入注意力机制增强关键特征识别能力使用残差连接缓解梯度消失问题最后接入全局平均池化层和全连接分类器输出故障类别概率。模型训练优化阶段初始化网络权重参数设置交叉熵损失函数和自适应矩估计优化器采用动态学习率调整策略在训练过程中实施梯度裁剪防止梯度爆炸通过前向传播计算预测输出反向传播更新网络参数迭代优化模型性能。模型验证与选择阶段在独立验证集上定期评估模型表现监控验证损失和分类准确率变化趋势保存验证性能最优的模型权重避免过拟合现象发生确保模型具备良好泛化能力。模型测试评估阶段加载最佳模型权重在未见过的测试集上进行全面性能评估计算总体分类准确率、每类故障的精确率和召回率生成混淆矩阵可视化分类错误分布定量分析模型诊断能力。参考文章基于时频谱图特征提取和改进型UNet卷积神经网络的机械故障诊断Pytorch - 哥廷根数学学派的文章 -https://zhuanlan.zhihu.com/p/1998402980043583749