o qi@s`UdZddlZddlmZdZdZdZe e d<dZ e e dfZ dZ d Zd ZGd d d ejjZdS) zhand written datasetN zmhttps://github.com/urduhack/resources/releases/download/urdu_handwritten_text_dataset_v1.0.0/uhat_dataset.npz _DOWNLOAD_URL i`i'c@s4eZdZdZejdZddZddZ ddZ d S) HandWrittenDigitszhand written Digitsz1.0.0c Cs:tjj|ttjtjjtdtjjt ddddt dS)N)shape) num_classesimagelabelz@https://www.kaggle.com/hazrat/uhat-urdu-handwritten-text-dataset)builder descriptionfeaturessupervised_keyshomepagecitation) tfdscore DatasetInfo _DESCRIPTIONr FeaturesDictImage _IMAGE_SHAPE ClassLabel _NUM_CLASSES _CITATION)selfrX/home/ubuntu/.local/lib/python3.10/site-packages/urduhack/datasets/image/hand_written.py_infos  zHandWrittenDigits._infocCsB|t}tjjtjj|ddddtjjtjj|ddddgS)zReturns SplitGenerators.x_digits_trainy_digits_train) file_pathr r )name gen_kwargs x_digits_test y_digits_test)downloadrrrSplitGeneratorSplitTRAINTEST)r dl_managerfilerrr_split_generators(s   z#HandWrittenDigits._split_generatorsc csft|}||}tj|dd}||}tt||}t|D]\}\} } | | d} || fVq dS)zYields examples.)axisr N)npload expand_dimslistzip enumerate) rr#r r urdu_datasetimageslabelsdataindeximgvaluerecordrrr_generate_examples7s   z$HandWrittenDigits._generate_examplesN) __name__ __module__ __qualname____doc__rrVersionVERSIONr r/r@rrrrrs   r)rDnumpyr2tensorflow_datasets.public_api public_apirrrrstr__annotations__ _IMAGE_SIZErr_TRAIN_EXAMPLES_TEST_EXAMPLESrGeneratorBasedBuilderrrrrrs