revert training and querying dataset