@@ -78,9 +78,9 @@ def get_imdb_data(vocabulary_size, max_len):
78
78
max_len: Cut text after this number of words.
79
79
80
80
Returns:
81
- x_train: An int array of shape `(num_exapmles , max_len)`: index-encoded
81
+ x_train: An int array of shape `(num_examples , max_len)`: index-encoded
82
82
sentences.
83
- y_train: An int array of shape `(num_exapmles ,)`: labels for the sentences.
83
+ y_train: An int array of shape `(num_examples ,)`: labels for the sentences.
84
84
x_test: Same as `x_train`, but for test.
85
85
y_test: Same as `y_train`, but for test.
86
86
"""
@@ -107,9 +107,9 @@ def train_model(model_type,
107
107
model_type: Type of the model to train, as a `str`.
108
108
vocabulary_size: Vocabulary size.
109
109
embedding_size: Embedding dimensions.
110
- x_train: An int array of shape `(num_exapmles , max_len)`: index-encoded
110
+ x_train: An int array of shape `(num_examples , max_len)`: index-encoded
111
111
sentences.
112
- y_train: An int array of shape `(num_exapmles ,)`: labels for the sentences.
112
+ y_train: An int array of shape `(num_examples ,)`: labels for the sentences.
113
113
x_test: Same as `x_train`, but for test.
114
114
y_test: Same as `y_train`, but for test.
115
115
epochs: Number of epochs to train the model for.
0 commit comments