From 7be70de65a114b2920beccddd4874778bdb4d378 Mon Sep 17 00:00:00 2001 From: qingfengtommy <52116238+qingfengtommy@users.noreply.github.com> Date: Wed, 2 Oct 2024 12:08:52 -0500 Subject: [PATCH] Fix PCA visualize issue.md --- episodes/7a-OOD-detection-output-based.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/episodes/7a-OOD-detection-output-based.md b/episodes/7a-OOD-detection-output-based.md index 1e55601f..77004757 100644 --- a/episodes/7a-OOD-detection-output-based.md +++ b/episodes/7a-OOD-detection-output-based.md @@ -121,7 +121,7 @@ def prep_ID_OOD_datasests(ID_class_labels, OOD_class_labels): test_labels = test_labels[test_filter] print(f'test_data.shape={test_data.shape}') - return ood_data, train_data, test_data + return ood_data, train_data, test_data, train_labels, test_labels def plot_data_sample(train_data, ood_data): @@ -151,7 +151,7 @@ def plot_data_sample(train_data, ood_data): ``` ```python -ood_data, train_data, test_data = prep_ID_OOD_datasests([0,1], [5]) +ood_data, train_data, test_data, train_labels, test_labels = prep_ID_OOD_datasests([0,1], [5]) fig = plot_data_sample(train_data, ood_data) fig.savefig('../images/OOD-detection_image-data-preview.png', dpi=300, bbox_inches='tight') plt.show()