if tag==EXIT: #MPI.Abort(MPI.COMM_WORLD) break comm.send(None, dest=0, tag=EXIT) if __name__ == "__main__": comm=MPI.COMM_WORLD size=comm.size rank=comm.rank status=MPI.Status() READY, START, DONE, EXIT = 0, 1, 2, 3 smile_old=zinc_data_with_bracket_original() val,smile=zinc_processed_with_bracket(smile_old) print val val2=['C', '(', ')', 'c', '1', '2', 'o', '=', 'O', 'N', '3', 'F', '[C@@H]', 'n', '-', '#', 'S', 'Cl', '[O-]', '[C@H]', '[NH+]', '[C@]', 's', 'Br', '/', '[nH]', '[NH3+]', '4', '[NH2+]', '[C@@]', '[N+]', '[nH+]', '\\', '[S@]', '5', '[N-]', '[n+]', '[S@@]', '[S-]', '6', '7', 'I', '[n-]', 'P', '[OH+]', '[NH-]', '[P@@H]', '[P@@]', '[PH2]', '[P@]', '[P+]', '[S+]', '[o+]', '[CH2-]', '[CH-]', '[SH+]', '[O+]', '[s+]', '[PH+]', '[PH]', '8', '[S@@+]'] logP_values = np.loadtxt('logP_values.txt') SA_scores = np.loadtxt('SA_scores.txt') cycle_scores = np.loadtxt('cycle_scores.txt') SA_mean = np.mean(SA_scores) #print len(SA_scores) SA_std=np.std(SA_scores) logP_mean = np.mean(logP_values) logP_std= np.std(logP_values) cycle_mean = np.mean(cycle_scores) cycle_std=np.std(cycle_scores)
print("depth=", depth) print(len(depth)) return valid_compound def UCTchemical(): state = chemical() state.Carbon() best = MCTS(root=state) return best if __name__ == "__main__": # os.environ['CUDA_VISIBLE_DEVICES'] = "" # use only CPU smile_old = zinc_data_with_bracket_original( ) # read 250k data as string into the list smile old val, smile = zinc_processed_with_bracket(smile_old) print(val) # val = ['\n', '&', 'C', '(', ')', 'c', '1', '2', 'o', '=', 'O', 'N', '3', 'F', '[C@@H]', 'n', '-', '#', 'S', 'Cl', '[O-]', '[C@H]', '[NH+]', '[C@]', 's', 'Br', '/', '[nH]', '[NH3+]', '4', '[NH2+]', '[C@@]', '[N+]', '[nH+]', '\\', '[S@]', '5', '[N-]', '[n+]', '[S@@]', '[S-]', '6', '7', 'I', '[n-]', 'P', '[OH+]', '[NH-]', '[P@@H]', '[P@@]', '[PH2]', '[P@]', '[P+]', '[S+]', '[o+]', '[CH2-]', '[CH-]', '[SH+]', '[O+]', '[s+]', '[PH+]', '[PH]', '8', '[S@@+]'] logP_values = np.loadtxt('logP_values.txt') SA_scores = np.loadtxt('SA_scores.txt') cycle_scores = np.loadtxt('cycle_scores.txt') SA_mean = np.mean(SA_scores) SA_std = np.std(SA_scores) logP_mean = np.mean(logP_values) logP_std = np.std(logP_values) cycle_mean = np.mean(cycle_scores) cycle_std = np.std(cycle_scores)