/behavioral_segmentation

behavioral segmentation program

Primary LanguagePython

BALとSODAのプログラムデータ

MAP上でドライブすることをテストとして学習する環境を作成した。 driving_env_segがBALのプログラムデータ driving_env_SODAがSODAのプログラムデータ

Getting started

1. pip install

使っているパッケージは以下です。

  • pip install tensorflow (or tensorflow-cpu or tensorflow-gpu)
  • pip install keras
  • pip install keras-rl
  • pip install gym
  • pip install numpy
  • pip install pygame

作成時のバージョン

  • windows 10
  • python 3.6.8
  • tensorflow-gpu 1.11.0
  • Keras 2.1.0
  • keras-rl 0.4.0
  • gym 0.17.1
  • numpy 1.17.5
  • pygame 1.9.6

3. Run the program

  • SODA-dqn.pyを実行で学習開始

  • BAL-main_learn_first_lay.pyを実行で1層目を学習、main.pyで2層目を学習