SaiVinay007

# Custom model
#---------------------------------------------------------------------------------
class CustomActorCritic(nn.Module):
    def __init__(self, input_dim, control_dim, hidden_dim, safety_dim, Num_Hidden_Shared, Num_Hidden_Control, Num_Hidden_Safety, std=0.0):
        super(CustomActorCritic, self).__init__()


        layers_shared_actor = []
        layers_safety_actor = []
        layers_control_actor = []

SaiVinay007

import gym
import gym_four
import numpy as np
from matplotlib import pyplot as plt
import seaborn as sns
'''
To change in step() of grid world
'''

SaiVinay007

# ~/.bashrc: executed by bash(1) for non-login shells.
# see /usr/share/doc/bash/examples/startup-files (in the package bash-doc)
# for examples

# If not running interactively, don't do anything
case $- in
    *i*) ;;
      *) return;;
esac

SaiVinay007

def Reduced_data_set(dataset,final_dimension, plot = False):
    '''
    This function is used to perform pca 
    and retain final_dimension number of 
    principle component from our dataset
    '''
    
    # Centering all the features to zero mean    
    Mean_of_data=dataset.mean(0)
    centered_data=dataset-Mean_of_data