o Me@sdZddlmZddlZddlZddlZddlZddlZddl Z ddl m m Z gZdZdZgdZdadadZdZd d Zdd dZe dddddddZe dddddddZddZe dddddddZe dddddddZdS)z UCI Housing dataset. This module will download dataset from https://archive.ics.uci.edu/ml/machine-learning-databases/housing/ and parse training set and test set into paddle reader creators. )print_functionNz:http://paddlemodels.bj.bcebos.com/uci_housing/housing.dataZ d4accdce7a25600298819f8e28e8d593) ZCRIMZZNZINDUSZCHASZNOXZRMZAGEZDISZRADZTAXZPTRATIOBZLSTATzYhttps://github.com/PaddlePaddle/book/raw/develop/01.fit_a_line/fluid/fit_a_line.fluid.tarZ 6e6dd637ccd5993961f68bfbde46090bcCsddl}|dddlm}|\}}t|}|jtt|||ddd| d| tt|t | d|g| d|d tjd sStd |jd d d ||dS)NrZAggrcenter)coloralignz feature scale z./imagezimage/ranges.png0)Zdpi) matplotlibZuseZmatplotlib.pyplotZpyplotZsubplotslenbarlistrange set_titleZxticks feature_namesZxlimZ set_figheightZ set_figwidthospathexistsmakedirsZsavefigclose)maximumsminimumsr ZpltZfigax feature_numrJD:\Projects\ConvertPro\env\Lib\site-packages\paddle/dataset/uci_housing.py feature_range0s&        r皙?c Cstdur tdur dStj|dd}||jd||}|jdd|jdd|jdd|jd}}}t j |dD]}|dd|f|||||||dd|f<q=t |jd|}|d|a||dadS)N )sepr)Zaxis) UCI_TRAIN_DATA UCI_TEST_DATAnpfromfileZreshapeshapemaxminsumsixmovesrint) filenamerratiodatarrZavgsioffsetrrr load_dataEs 6 r4z2.0.0zpaddle.text.datasets.UCIHousingr#z>Please use new dataset API which supports paddle.io.DataLoader)ZsinceZ update_tolevelreasoncC"ttjjtdtdd}|S)z UCI_HOUSING training set creator. It returns a reader creator, each sample in the reader is features after normalization and price number. :return: Training reader creator :rtype: callable uci_housingcs*tD]}|dd|ddfVqdSNr)r$drrrreaderiztrain..readerr4paddledatasetcommondownloadURLMD5r=rrrtrainWrGcCr7)z UCI_HOUSING test set creator. It returns a reader creator, each sample in the reader is features after normalization and price number. :return: Test reader creator :rtype: callable r8csr9r:)r%r;rrrr=r>ztest..readerr?rFrrrtestprHrIcCs:tjjtdtd}tj|dd}t }|j |d|S)Nr8zfit_a_line.fluid.tarr)mode)r) r@rArBrCFLUID_URL_MODELFLUID_MD5_MODELtarfileTarFiletempfilemkdtemp extractall)Z parameter_tartardirpathrrr fluid_models  rTcCs(ttjjtdttdddfS)zh It returns just one tuple data to do inference. :return: one tuple data :rtype: tuple r8rNr)r4r@rArBrCrDrEr%rrrrpredict_readers rUcCstjjtdtdS)Nr8)r@rArBrCrDrErrrrfetchsrV)rr )__doc__ __future__rnumpyr&r,rOrMrZpaddle.dataset.commonr@Zpaddle.utils.deprecatedutils deprecated__all__rDrErr$r%rKrLrr4rGrIrTrUrVrrrrs`