def visualize_umap_data()

in torchbenchmark/models/dlrm/tools/visualize.py [0:0]

• 117 lines of code
• 1 McCabe index (conditional complexity)

def visualize_umap_data(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):

    # all classes
    plot_all_data_3(umap_Y           = umap_Y,
                    umap_T           = umap_T,
                    train_Y          = train_Y,
                    train_T          = train_T, 
                    test_Y           = test_Y, 
                    test_T           = test_T, 
                    total_train_size = total_train_size,
                    total_test_size  = total_test_size,
                    info             = info,
                    output_dir       = output_dir,
                    orig_space_dim   = orig_space_dim)

    # all predictions
    plot_all_data_3(umap_Y           = umap_Y,
                    umap_T           = umap_P,
                    train_Y          = train_Y,
                    train_T          = train_P, 
                    test_Y           = test_Y, 
                    test_T           = test_P, 
                    total_train_size = total_train_size,
                    total_test_size  = total_test_size,
                    info             = info+", all-predictions",
                    output_dir       = output_dir,
                    orig_space_dim   = orig_space_dim)

    
    # class 0
    plot_one_class_3(umap_Y           = umap_Y,
                     umap_T           = umap_T,
                     train_Y          = train_Y,
                     train_T          = train_T,
                     test_Y           = test_Y, 
                     test_T           = test_T, 
                     target           = 0, 
                     col              = "red", 
                     total_train_size = total_train_size, 
                     total_test_size  = total_test_size, 
                     info             = info+" class " + str(0),
                     output_dir       = output_dir,
                     orig_space_dim   = orig_space_dim)

    # class 1
    plot_one_class_3(umap_Y           = umap_Y,
                     umap_T           = umap_T,
                     train_Y          = train_Y,
                     train_T          = train_T,
                     test_Y           = test_Y, 
                     test_T           = test_T, 
                     target           = 1, 
                     col              = "green", 
                     total_train_size = total_train_size, 
                     total_test_size  = total_test_size, 
                     info             = info + " class " + str(1),
                     output_dir       = output_dir,
                     orig_space_dim   = orig_space_dim)

    # correct classification
    plot_one_class_3(umap_Y           = umap_Y,
                     umap_T           = umap_C,
                     train_Y          = train_Y,
                     train_T          = train_C,
                     test_Y           = test_Y, 
                     test_T           = test_C, 
                     target           = 0, 
                     col              = "green", 
                     total_train_size = total_train_size, 
                     total_test_size  = total_test_size, 
                     info             = info + " correct ",
                     output_dir       = output_dir,
                     orig_space_dim   = orig_space_dim)

    # errors
    plot_one_class_3(umap_Y           = umap_Y,
                     umap_T           = umap_C,
                     train_Y          = train_Y,
                     train_T          = train_C,
                     test_Y           = test_Y, 
                     test_T           = test_C, 
                     target           = 1, 
                     col              = "red", 
                     total_train_size = total_train_size, 
                     total_test_size  = total_test_size, 
                     info             = info + " errors ",
                     output_dir       = output_dir,
                     orig_space_dim   = orig_space_dim)

    # prediction 0
    plot_one_class_3(umap_Y           = umap_Y,
                     umap_T           = umap_P,
                     train_Y          = train_Y,
                     train_T          = train_P,
                     test_Y           = test_Y, 
                     test_T           = test_P, 
                     target           = 0, 
                     col              = "red", 
                     total_train_size = total_train_size, 
                     total_test_size  = total_test_size, 
                     info             = info + " predict-0 ",
                     output_dir       = output_dir,
                     orig_space_dim   = orig_space_dim)

    # prediction 1
    plot_one_class_3(umap_Y           = umap_Y,
                     umap_T           = umap_P,
                     train_Y          = train_Y,
                     train_T          = train_P,
                     test_Y           = test_Y, 
                     test_T           = test_P, 
                     target           = 1, 
                     col              = "green", 
                     total_train_size = total_train_size, 
                     total_test_size  = total_test_size, 
                     info             = info + " predict-1 ",
                     output_dir       = output_dir,
                     orig_space_dim   = orig_space_dim)