def score_function_custom_maker(filename='trajectory.hdf5', decay=4): """ param: trajectory file with key "filled_path" decay """ with h5py.File(filename, 'r') as file: filled_path = file['filled_path'][:] file.close() score_function = trajectory_to_score_function.score_function_maker(filled_path.T, decay) return score_function
if test_trajectory_to_score_function: import triple_well_2D.triple_well as tw import triple_well_2D.tools_2D as too """ input: trajectory transverse decay """ #arguments trajectory = positions decay = decay sigma = noise #result score_function = trajectory_to_score_function.score_function_maker( trajectory.T, decay) #visualise too.visualise_scorefunction(score_function=score_function, sigma=sigma, target_state=tw.target_state, score_threshold=None, force_matrix=tw.force_matrix, xmin=-4, xmax=4, ymin=-3, ymax=6) #%%7 if test_warp_score_function_ell: import triple_well_2D.triple_well as tw
plt.legend(loc='lower right') plt.savefig('../../Report/overleaf/low_noise_histo', bbox_inches='tight') #plt.show() if write: with h5py.File(filename, 'a') as file: file.create_dataset('original_path', data=path) file.create_dataset('filled_path', data=filled_path) file.close() #%% if check_score_function: decay = 2 score_function = trajectory_to_score_function.score_function_maker( filled_path.T, decay) #visualise plt.figure() print(len(filled_path[0])) #filled_path = histogram_to_trajectory.positions_filler(filled_path, 100) print(len(filled_path[0])) plt.scatter(filled_path[0], filled_path[1], label='path', color='red', zorder=10, s=3) #plt.text(-0.2, 1,'(a)',horizontalalignment='center', verticalalignment='center', transform = plt.gca().transAxes, fontsize=9) too.visualise_scorefunction(score_function=score_function, initial_state=tw.initial_state,