from std_msgs.msg import Int32
from time import sleep
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0,3"
rospy.init_node('pub')
pub = rospy.Publisher('car', Int32, queue_size=20)
scores = []
print()
print("run")
print()
if __name__ == '__main__':
global_step = 0
agent = dqn(365, 5, 100000000, 0.001) # state size 365, output size 5
env = Env()
#r = rospy.Duration(1,0)
for e in range(1, 3000):
done = False
state = env.reset()
score = 0
print()
print("run")
print()
for t in range(1, 3000):
print()
print("run")
print()
action = agent.get_action(state)
pub.publish(action) # robot run
rospy.sleep(10.)
###global setting xscale = 'liner'#ACCEPT [integer 1 to 10 | 'liner'] example xscale = 10 #means xscale use log10 yscale = 'liner' ####solve data file1 workpath = r"F:\Users\JianyingLi\Desktop\S1P0.05E1" indexname = "eRHo" extname = ".dat" timeCoe = 10e-7 lengthCoe = 0.0003125 instance = Env() print instance.get_workingPath() fileReader = filereader(instance.get_fileList(workpath), workpath) fileReader.select_files(extname, indexname) fileReader.read_data() # print fileReader.data dp = DataProcess() x0,y0 = dp.data1D(fileReader.data, extname, indexname,timeCoe) ##################################################################### ####solve data file2 workpath = r"F:\Users\JianyingLi\Desktop\S1P0.05E0.4" indexname = "eRHo"