train_mnist.py

#!/bin/python
#-----------------------------------------------------------------------------
# File Name : MNIST_SRM_RATE_script200.py
# Purpose:
#
# Author: Emre Neftci
#
# Creation Date : 01-10-2013
# Last Modified : Fri 27 Jun 2014 02:44:18 PM PDT
#
# Copyright : (c) UCSD, Emre Neftci, Srinjoy Das, Bruno Pedroni, Kenneth Kreutz-Delgado, Gert Cauwenberghs
# Licence : GPLv2
#----------------------------------------------------------------------------- 

import meta_parameters
meta_parameters.parameters_script = 'parameters_mnist'

from MNIST_IF_STDP import *

if __name__ == '__main__':
    mnist_data = load_mnist_data(min_p = .00001, max_p = .98, binary = True, seed=0)
    Id = create_Id(mnist_data, c_min_p = .00001, c_max_p = .98)
    W, b_v, b_c, b_h = create_rbm_parameters()
    loc = main(W, b_v, b_c, b_h, Id = Id, dorun = True, monitors = False, display=True, mnist_data=mnist_data)
    locals().update(loc)
    W=Srs.w.data.copy().reshape(N_v+N_c, N_h)*beta
    Wvh=W[:N_v,:]
    Wch=W[N_v:,:]
    mBv = Sbv.w.data*beta*tau_rec*bias_input_rate
    mBh = Sbh.w.data*beta*tau_rec*bias_input_rate
    b_c = mBv[N_v:(N_v+N_c)]
    b_v = mBv[:N_v]
    b_h = mBh
    mB = np.concatenate([mBv,mBh])

    d = et.mksavedir()
    et.save_file(__file__)
    et.globaldata.W = W
    et.globaldata.mB = mB
    
    try:
        et.globaldata.Mv = monitor_to_spikelist(Mv)
        et.globaldata.Mh = monitor_to_spikelist(Mh)
    except NameError:
        print "SpikeMonitors are not defined"
        
    et.globaldata.res_hist_train = res_hist_train 
    et.globaldata.res_hist_test = res_hist_test 
    et.globaldata.w_hist_v   = w_hist_v
    et.globaldata.w_hist_c   = w_hist_c
    et.globaldata.b_hist_vcn = res_hist_train
    et.globaldata.b_hist_h   = res_hist_test 
    et.save({'Wh':Wvh, 'Wc':Wch, 'b_vch': mB}, 'WSCD.pkl')
    et.save()
    et.savefig('progress.png', format='png')
#