Example #1
0
def augment_state(state):
    #import pdb
    #pdb.set_trace()
    # TODO: Move into Definitions + Add propapagation
    return tf.concat(
        to_col([State.TFP(state), State.depr(state)] +
               [getattr(State, "K" + str(i))(state) for i in range(2, 7)] + [
                   Definitions.K_total(state, None),
                   Definitions.r(state, None),
                   Definitions.w(state, None),
                   Definitions.Y(state, None)
               ] + [
                   Definitions.r(state, None) *
                   getattr(State, "K" + str(i))(state) for i in range(2, 7)
               ]),
        axis=1)
def w(state, policy_state=None):
    return (1 - Parameters.alpha) * State.TFP(state) * Definitions.K_total(
        state, None)**Parameters.alpha
def Y(state, policy_state=None):
    return State.TFP(state) * Definitions.K_total(
        state, None)**Parameters.alpha + (
            1 - State.depr(state)) * Definitions.K_total(state, None)
def r(state, policy_state=None):
    return Parameters.alpha * State.TFP(state) * Definitions.K_total(
        state, None)**(Parameters.alpha - 1) + (1 - State.depr(state))