Skip to content

Instantly share code, notes, and snippets.

View jwjohnson314's full-sized avatar

Jeremiah Johnson jwjohnson314

View GitHub Profile
@jwjohnson314
jwjohnson314 / pg-pong.py
Last active June 6, 2016 16:40 — forked from karpathy/pg-pong.py
Training a Neural Network ATARI Pong agent with Policy Gradients from raw pixels
#!/usr/bin/env python
"""
Trains an agent with (stochastic) Policy Gradients on Pong. Uses OpenAI Gym.
Python3 port
"""
import numpy as np
import pickle
import gym
# hyperparameters