Skip to content

Instantly share code, notes, and snippets.

@joshreini1
Created October 7, 2022 20:06
Show Gist options
  • Save joshreini1/4bc1d50c4195a764cc279ca0f98dfba5 to your computer and use it in GitHub Desktop.
Save joshreini1/4bc1d50c4195a764cc279ca0f98dfba5 to your computer and use it in GitHub Desktop.
data_collections = tru.get_data_collections()
for dc in data_collections:
tru.set_data_collection(dc)
splits = tru.get_data_splits()
for split_key in splits:
split = f'{split_key}'
tru.tester.add_stability_test(
comparison_data_split_name = split,
base_data_split_name = 'train',
metric = 'DIFFERENCE_OF_MEAN',
warn_if_outside=[-.05, .05],
fail_if_outside=[-.1, 1],
overwrite = True
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment