Python 保存和加载自动编码器_Python_Neural Network_Pyspark_Prediction_Autoencoder

Python 保存和加载自动编码器

python neural-network pyspark

Python 保存和加载自动编码器,python,neural-network,pyspark,prediction,autoencoder,Python,Neural Network,Pyspark,Prediction,Autoencoder,我正在尝试使用自动编码器（AE）和交替最小二乘法（ALS）方法制作预测值。 ALS是使用pyspark.mllib.recommendation包创建的。我可以通过model.save（）和pyspark.mllib.recommendation.MatrixFactoryModel.load（）方法保存ALS模型并重用它 AE是使用torch.nn.Module包创建的，有4层。与ALS不同，AE保存和加载模型并不是我所缺少的。有人能帮我找到保存和加载自动编码器的方法吗。torch模块提供保存

我正在尝试使用自动编码器（AE）和交替最小二乘法（ALS）方法制作预测值。 ALS是使用pyspark.mllib.recommendation包创建的。我可以通过model.save（）和pyspark.mllib.recommendation.MatrixFactoryModel.load（）方法保存ALS模型并重用它

AE是使用torch.nn.Module包创建的，有4层。与ALS不同，AE保存和加载模型并不是我所缺少的。有人能帮我找到保存和加载自动编码器的方法吗。

torch模块提供保存和加载方法

它将模型保存为.pth文件

torch.save(model,'model.pth')

它还具有加载方法

torch.load(.pth_file)