MCPcopy Index your code
hub / github.com/facebookresearch/dlrm / visualize_umap_data

Function visualize_umap_data

tools/visualize.py:611–760  ·  view source on GitHub ↗
(
    umap_Y,
    umap_T,
    umap_C,
    umap_P,
    train_Y,
    train_T,
    train_C,
    train_P,
    test_Y=None,
    test_T=None,
    test_C=None,
    test_P=None,
    total_train_size="",
    total_test_size="",
    info="",
    output_dir="",
    orig_space_dim=0,
)

Source from the content-addressed store, hash-verified

609
610
611def visualize_umap_data(
612 umap_Y,
613 umap_T,
614 umap_C,
615 umap_P,
616 train_Y,
617 train_T,
618 train_C,
619 train_P,
620 test_Y=None,
621 test_T=None,
622 test_C=None,
623 test_P=None,
624 total_train_size="",
625 total_test_size="",
626 info="",
627 output_dir="",
628 orig_space_dim=0,
629):
630 # all classes
631 plot_all_data_3(
632 umap_Y=umap_Y,
633 umap_T=umap_T,
634 train_Y=train_Y,
635 train_T=train_T,
636 test_Y=test_Y,
637 test_T=test_T,
638 total_train_size=total_train_size,
639 total_test_size=total_test_size,
640 info=info,
641 output_dir=output_dir,
642 orig_space_dim=orig_space_dim,
643 )
644
645 # all predictions
646 plot_all_data_3(
647 umap_Y=umap_Y,
648 umap_T=umap_P,
649 train_Y=train_Y,
650 train_T=train_P,
651 test_Y=test_Y,
652 test_T=test_P,
653 total_train_size=total_train_size,
654 total_test_size=total_test_size,
655 info=info + ", all-predictions",
656 output_dir=output_dir,
657 orig_space_dim=orig_space_dim,
658 )
659
660 # class 0
661 plot_one_class_3(
662 umap_Y=umap_Y,
663 umap_T=umap_T,
664 train_Y=train_Y,
665 train_T=train_T,
666 test_Y=test_Y,
667 test_T=test_T,
668 target=0,

Callers 1

visualize_all_data_umapFunction · 0.85

Calls 2

plot_all_data_3Function · 0.85
plot_one_class_3Function · 0.85

Tested by

no test coverage detected