U C^@s6ddlmZddlmZGdddeZd ddZdS) )unicode_literals)Modelc@s&eZdZddZd ddZddZdS) FeatureExtractercCst|||_dSN)r__init__attrs)selfrr J/tmp/pip-install-6_kvzl1k/thinc/thinc/neural/_classes/feature_extracter.pyrs zFeatureExtracter.__init__csfdd|D}|tfS)Ncsg|]}|qSr ) _get_feats).0docr r r sz1FeatureExtracter.begin_update..)_feature_extracter_bwd)r ZdocsZdropfeaturesr rr begin_update szFeatureExtracter.begin_updatecCsBt|dr||j}n|j|j|j|j}|jj|ddS)Nto_arrayZuint64)Zdtype)hasattrrrrstartendopsZasarray)r rZarrr r r r s zFeatureExtracter._get_featsN)r )__name__ __module__ __qualname__rrr r r r r rs rNcCs|Srr )Z d_featuresZsgdr r r rsr)N) __future__rmodelrrrr r r r s