Linux安装并使用Ludwig训练模型 – Python量化投资

Linux安装并使用Ludwig训练模型

一、在Python3环境安装Ludwig:

  1、使用pip安装:pip install ludwig

  2、源码编译安装:

    git clone git@github.com:uber/ludwig.git

    cd ludwig

    pip install -r requirements.txt

    python setup.py install

  3、准备csv格式的数据,此处以 reuters-allcats.csv 为示例

二、安装特定语言的模型:

  python -m spacy download en

注:需要安装libsndfile1(如环境中已有则忽略):sudo apt-get install libsndfile1

三、下一步将是配置模型定义YAML文件,该文件指定模型的输入和输出。(以下为yaml文件内容)

  input_features:
      –
          name: text
          type: text
          encoder: parallel_cnn
          level: word
  
  output_features:
      –
          name: class
          type: category

四、在成功安装完成之后,执行命令,开始试验训练模型,即可已默认参数进行模型的训练:

  ludwig experiment –data_csv reuters-allcats.csv –model_definition_file model_definition.yaml

 

Ludwig的其他功能,包括根据模型的输出日志进行可视化分析,模型预测,分布式训练以及其API等。

https://www.cnblogs.com/qcly/p/qcly-ludwig.html

「点点赞赏,手留余香」

    还没有人赞赏,快来当第一个赞赏的人吧!
0 条回复 A 作者 M 管理员
    所有的伟大,都源于一个勇敢的开始!
欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论