train data

Windows/mlp_model.py

@@ -108,11 +108,27 @@ class MLPClassifier:
         self.train()
         self.save_model()
 
+    def predict(self, X):
+        self.load_model()  # Ensure the model parameters are loaded
+        with torch.no_grad():
+            y_hat = self.net(X)
+            _, predicted = torch.max(y_hat, 1)
+        return predicted
+
+
 
 if __name__ == '__main__':
     mlp_classifier = MLPClassifier()
     file_path='trained_mlp.pkl'
-    mlp_classifier.run()
+    if not os.path.exists(file_path):
+     mlp_classifier.run()
+    test_data = torch.tensor([[97,58], [25,34], [42,7],[298,9],[10,39]], dtype=torch.float32)
+
+    # 进行预测
+    predictions = mlp_classifier.predict(test_data)
+
+    # 输出预测结果
+    print("Predictions:", predictions.numpy())