why not create a function inside the Dataset object, something like update which might take arguments required and updates the data[‘prob’]. You will have have freedom to call the function when you want (after how many cross validations you want…)
I believe this should work, have done something similar myself once