新建仓库维护数据预测项目

FC_ML_NN_Model/Poly_Model.py

@@ -0,0 +1,10 @@
+import torch
+
+
+class PolyModel(torch.nn.Module):
+    def __init__(self,input_size,output_size):
+        super().__init__()
+        self.linear = torch.nn.Linear(input_size, output_size)  # 输入n维，输出m维
+
+    def forward(self, x):
+        return self.linear(x)