当前位置:   article > 正文

深度学习-如何调试基于spark的LSTM_spark的lstm算法

spark的lstm算法

本文是在spark本地模式运行,如果想用集群模式,使用上篇提到的参数即可,例子是LSTM字符,代码如下

public class TrainingStatsExample {
    private static final Logger log = LoggerFactory.getLogger(TrainingStatsExample.class);

    @Parameter(names="-useSparkLocal", description = "Use spark local (helper for testing/running without spark submit)", arity = 1)//设置参数名,描述,使用参数数量
    private boolean useSparkLocal = true;//设置参数值,使用本地模式

    public static void main(String[] args) throws Exception {
        new TrainingStatsExample().entryPoint(args);//传入参数,调用入口函数
    }

    private void entryPoint(String[] args) throws Exception {
        //Handle command line arguments
        JCommander jcmdr = new JCommander(this);//处理命令行的类
        try{
            jcmdr.parse(args);//解析
        } catch(ParameterException e){
            //User provides invalid input -> print the usage info
            jcmdr.usage();//如果是无效输入,打印错误
            try{ Thread.sleep(500); } catch(Exception e2){ }
            throw e;
        }


        //Set up network configuration:
        MultiLayerConfiguration config = getConfiguration(
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/IT小白/article/detail/359440
推荐阅读
相关标签
  

闽ICP备14008679号