joblib.dump(model, 'model_train.pkl')将这行代码生成的文件上传到web端并提供下载链接的python代码和html代码
时间: 2024-05-16 17:15:06 浏览: 194
Python代码:
```python
from flask import Flask, send_file
import joblib
app = Flask(__name__)
# 定义路由,当访问/download时,执行以下函数
@app.route('/download')
def download_file():
# 加载模型文件
model = joblib.load('model_train.pkl')
# 将模型文件保存在服务器上的文件夹中
model_path = 'model_train.pkl'
# 返回文件下载链接
return send_file(model_path, as_attachment=True)
if __name__ == '__main__':
app.run()
```
HTML代码:
```html
<!DOCTYPE html>
<html>
<head>
<title>Download Model</title>
</head>
<body>
<h1>Download Model</h1>
<a href="{{ url_for('download_file') }}">Download Model</a>
</body>
</html>
```
在Flask应用程序中,我们定义了一个/download路由,当用户访问/download时,会执行download_file函数,该函数会加载模型文件,将其保存在服务器上的文件夹中,并返回模型文件的下载链接。在HTML代码中,我们创建一个链接,指向/download路由,当用户单击链接时,会下载模型文件。
阅读全文