(
umap_Y,
umap_T,
train_Y=None,
train_T=None,
test_Y=None,
test_T=None,
total_train_size="",
total_test_size="",
info="",
output_dir="",
orig_space_dim=0,
)
| 491 | |
| 492 | |
| 493 | def plot_all_data_3( |
| 494 | umap_Y, |
| 495 | umap_T, |
| 496 | train_Y=None, |
| 497 | train_T=None, |
| 498 | test_Y=None, |
| 499 | test_T=None, |
| 500 | total_train_size="", |
| 501 | total_test_size="", |
| 502 | info="", |
| 503 | output_dir="", |
| 504 | orig_space_dim=0, |
| 505 | ): |
| 506 | size = 1 |
| 507 | colors = ["red", "green"] |
| 508 | |
| 509 | fig, (ax0, ax1, ax2) = plt.subplots(1, 3) |
| 510 | fig.suptitle("UMAP: " + info + " space dim " + str(orig_space_dim)) |
| 511 | |
| 512 | ax0.scatter( |
| 513 | umap_Y[:, 0], |
| 514 | umap_Y[:, 1], |
| 515 | s=size, |
| 516 | c=umap_T, |
| 517 | cmap=matplotlib.colors.ListedColormap(colors), |
| 518 | marker=".", |
| 519 | linewidth=0, |
| 520 | ) |
| 521 | ax0.set_title( |
| 522 | "UMAP (" + str(len(umap_T)) + " of " + total_train_size + ")", fontsize=7 |
| 523 | ) |
| 524 | |
| 525 | if train_Y is not None and train_T is not None: |
| 526 | ax1.scatter( |
| 527 | train_Y[:, 0], |
| 528 | train_Y[:, 1], |
| 529 | s=size, |
| 530 | c=train_T, |
| 531 | cmap=matplotlib.colors.ListedColormap(colors), |
| 532 | marker=".", |
| 533 | linewidth=0, |
| 534 | ) |
| 535 | ax1.set_title( |
| 536 | "Train (" + str(len(train_T)) + " of " + total_train_size + ")", fontsize=7 |
| 537 | ) |
| 538 | |
| 539 | if test_Y is not None and test_T is not None: |
| 540 | ax2.scatter( |
| 541 | test_Y[:, 0], |
| 542 | test_Y[:, 1], |
| 543 | s=size, |
| 544 | c=test_T, |
| 545 | cmap=matplotlib.colors.ListedColormap(colors), |
| 546 | marker=".", |
| 547 | linewidth=0, |
| 548 | ) |
| 549 | ax2.set_title( |
| 550 | "Test (" + str(len(test_T)) + " of " + total_test_size + ")", fontsize=7 |
no outgoing calls
no test coverage detected