Skip to content

Commit 27504e6

Browse files
authored
Fix the vocabulary_dtype in keras.layers.IntegerLookup() (#21587)
1 parent db409ff commit 27504e6

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

keras/src/layers/preprocessing/integer_lookup.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,8 +76,9 @@ class IntegerLookup(IndexLookup):
7676
If passing a file path, the file should contain one line per term
7777
in the vocabulary. If this argument is set,
7878
there is no need to `adapt()` the layer.
79-
vocabulary_dtype: The dtype of the vocabulary terms, for example
80-
`"int64"` or `"int32"`. Defaults to `"int64"`.
79+
vocabulary_dtype: The dtype of the vocabulary terms.
80+
Only `vocabulary_dtype='int64'` is supported at this time.
81+
Defaults to `"int64"`.
8182
idf_weights: Only valid when `output_mode` is `"tf_idf"`.
8283
A tuple, list, 1D NumPy array, or 1D tensor or the same length
8384
as the vocabulary, containing the floating point inverse document

0 commit comments

Comments
 (0)