experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Export text_dataset_from_directory to the public API.

Browse Source 
PiperOrigin-RevId: 308115446
Change-Id: I37cce67a9560906560b816ca387ee86644740ded
This commit is contained in:
Francois Chollet 2020-04-23 13:17:43 -07:00 committed by TensorFlower Gardener
parent 3ce6c4f86f
commit 43832e2a10
2 changed files with 6 additions and 0 deletions
tensorflow
python/keras/preprocessing
text_dataset.py
tools/api/golden/v2
tensorflow.keras.preprocessing.pbtxt

2
tensorflow/python/keras/preprocessing/text_dataset.py
View File

@ -23,8 +23,10 @@ from tensorflow.python.data.ops import dataset_ops
from tensorflow.python.keras.preprocessing import dataset_utils
from tensorflow.python.ops import io_ops
from tensorflow.python.ops import string_ops
from tensorflow.python.util.tf_export import keras_export
@keras_export('keras.preprocessing.text_dataset_from_directory', v1=[])
def text_dataset_from_directory(directory,
labels='inferred',
label_mode='int',

4
tensorflow/tools/api/golden/v2/tensorflow.keras.preprocessing.pbtxt
View File

@ -16,6 +16,10 @@ tf_module {
name: "image_dataset_from_directory"
argspec: "args=[\'directory\', \'labels\', \'label_mode\', \'class_names\', \'color_mode\', \'batch_size\', \'image_size\', \'shuffle\', \'seed\', \'validation_split\', \'subset\', \'interpolation\', \'follow_links\'], varargs=None, keywords=None, defaults=[\'inferred\', \'int\', \'None\', \'rgb\', \'32\', \'(256, 256)\', \'True\', \'None\', \'None\', \'None\', \'bilinear\', \'False\'], "
}
member_method {
name: "text_dataset_from_directory"
argspec: "args=[\'directory\', \'labels\', \'label_mode\', \'class_names\', \'batch_size\', \'max_length\', \'shuffle\', \'seed\', \'validation_split\', \'subset\', \'follow_links\'], varargs=None, keywords=None, defaults=[\'inferred\', \'int\', \'None\', \'32\', \'None\', \'True\', \'None\', \'None\', \'None\', \'False\'], "
}
member_method {
name: "timeseries_dataset_from_array"
argspec: "args=[\'data\', \'targets\', \'sequence_length\', \'sequence_stride\', \'sampling_rate\', \'batch_size\', \'shuffle\', \'seed\', \'start_index\', \'end_index\'], varargs=None, keywords=None, defaults=[\'1\', \'1\', \'128\', \'False\', \'None\', \'None\', \'None\'], "