|
630 | 630 | "will include original sample features as well as features of their corresponding\n", |
631 | 631 | "neighbors.\n", |
632 | 632 | "\n", |
633 | | - "In this tutorial, we consider undirected edges and we use a maximum of 1\n", |
634 | | - "neighbor per sample." |
| 633 | + "In this tutorial, we consider undirected edges and use a maximum of 3 neighbors\n", |
| 634 | + "per sample to augment training data with graph neighbors." |
635 | 635 | ] |
636 | 636 | }, |
637 | 637 | { |
|
659 | 659 | "source": [ |
660 | 660 | "## Base model\n", |
661 | 661 | "\n", |
662 | | - "We are now ready to build a base model without graph regularization. In order to build this model, we can either use embeddings that were used in building the graph, or we can learn new embeddings jointly along with the classification task. For the purpose of this notebook, we will do the latter." |
| 662 | + "We are now ready to build a base model without graph regularization. In order to\n", |
| 663 | + "build this model, we can either use embeddings that were used in building the\n", |
| 664 | + "graph, or we can learn new embeddings jointly along with the classification\n", |
| 665 | + "task. For the purpose of this notebook, we will do the latter." |
663 | 666 | ] |
664 | 667 | }, |
665 | 668 | { |
|
701 | 704 | "\n", |
702 | 705 | "- **num_classes**: There are 2 classes -- *positive* and *negative*.\n", |
703 | 706 | "\n", |
704 | | - "- **max_seq_length**: This is the maximum number of words considered from each movie review in this example.\n", |
| 707 | + "- **max_seq_length**: This is the maximum number of words considered from each\n", |
| 708 | + " movie review in this example.\n", |
705 | 709 | "\n", |
706 | | - "- **vocab_size**: This is the size of the vocabulary considered for this example.\n", |
| 710 | + "- **vocab_size**: This is the size of the vocabulary considered for this\n", |
| 711 | + " example.\n", |
707 | 712 | "\n", |
708 | 713 | "- **distance_type**: This is the distance metric used to regularize the sample\n", |
709 | 714 | " with its neighbors.\n", |
|
712 | 717 | " the graph regularization term in the overall loss function.\n", |
713 | 718 | "\n", |
714 | 719 | "- **num_neighbors**: The number of neighbors used for graph regularization.\n", |
| 720 | + " This value has to be less than or equal to the `--max_nbrs` command line\n", |
| 721 | + " value used above when you ran the `pack_nbrs` utility.\n", |
715 | 722 | "\n", |
716 | | - "- **num_fc_units**: The number of units in the fully connected layer of the neural network.\n", |
| 723 | + "- **num_fc_units**: The number of units in the fully connected layer of the\n", |
| 724 | + " neural network.\n", |
717 | 725 | "\n", |
718 | 726 | "- **train_epochs**: The number of training epochs.\n", |
719 | 727 | "\n", |
|
0 commit comments