Skip to content

Instantly share code, notes, and snippets.

@erykml

erykml/data_validation.py

Created May 1, 2022 14:03
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save erykml/248a6599236fab145e0ab5da4c35d90c to your computer and use it in GitHub Desktop.
Save erykml/248a6599236fab145e0ab5da4c35d90c to your computer and use it in GitHub Desktop.
Download ZIP
Raw
data_validation.py
from deepchecks.vision.simple_classification_data import load_dataset
from deepchecks.vision.suites import train_test_validation
from config import PROCESSED_IMAGES_DIR
train_ds = load_dataset(PROCESSED_IMAGES_DIR, train=True, object_type="VisionData", image_extension="jpg")
test_ds = load_dataset(PROCESSED_IMAGES_DIR, train=False, object_type="VisionData", image_extension="jpg")
suite = train_test_validation()
result = suite.run(train_ds, test_ds)
result.save_as_html("data_validation.html")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment