将Spark的ML/MLlib机器学习库中算法生成的模型存入数据库

在使用Spark的ML/MLlib过程中,想要将算法生成的模型进行保存,方便下次调用,模型的save方法可以将模型以文件的形式保存到磁盘中,但是如果代码运行在其他环境中想要调用模型的话,需要将模型文件copy到其他环境中并配置好模型文件的路径,这样就很麻烦。所以将模型保存至数据库中,其他环境调用起来就很方便。

解决思路

写入:将模型转换为二进制流存入数据库
读取:将数据库中读取的数据进行反序列化,强制转换为对应的模型类型

具体实现

表:

CREATE TABLE `tb_sms_model` (
  `id` varchar(255) NOT NULL,
  `model` blob COMMENT '模型',
  `create_time` datetime DEFAULT NULL COMMENT '创建时间',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8;

模型对象(.java):

public class SensitiveSMSModel implements Serializable {

    private String id;// 主键id

    private byte[] model; // 模型

    private String createTime;// 创建时间

    public String getId() {
        return id;
    }

    public void setId(String id) {
        this.id = id;
 
  • 4
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值