Giter Club home page Giter Club logo

stockkiller's Introduction

Big homework for Parttern Recognition

Created by Yufei Huang and Jaewon Kim

股市术语

  • Turover:成交额
  • Volume:成交量
  • Bid Price:买家出的价
  • Ask Price:卖家出的价

环境配置

  • Python3

  • pytorch

  • sklearn

  • matplotlib

  • pandas

参数设置

模型的参数设置在train.py文件的15行到38行,可以通过修改args对应的数据进行修改

# parameters
args = dict()
args['predict_len'] = 120 # 10min
args['epoch'] = 100
args['learning_rate'] = 0.0001
args['batch_size'] = 2000
args['lr_decay_factor'] = 0.9
args['input_dim'] = 1
args['hidden_size'] = 100
args['num_layers'] = 2
args['a'] = 30
args['b'] = 300
args['dt'] = 5
args['k'] = 0.3
args['theta'] = 0.004
args['save_path'] = './models_'+'pl'+str(args['predict_len'])+'_lr'+str(args['learning_rate'])+'_hd'+str(args['hidden_size'])
args['load_path'] = 'model6.pt'
args['step_size'] = 1000
args['load_model'] = True
args['gpu'] = 'cuda:3'
args['isTrain'] = False
args['imagepath'] = "./Image_" + 'pl'+str(args['predict_len'])+'_lr'+str(args['learning_rate'])+'_hd'+str(args['hidden_size'])
args['path'] = './PRdata'

运行方法

将数据文件PRData放置在指定目录,PRData中应包含至少三个文件夹TickOrderOrderQueue,运行程序后会在改数据目录中新建目录HandleTick存放已经预处理好的数据。

python3 train.py

注:如果采用GPU,注意需要GPU内存至少2G

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.