o èp©iã@s¸ddlmZmZddlmZddlmZddlmZGdd„dƒZ Gdd „d e ƒZ Gd d „d e ƒZ Gd d „d e ƒZ Gdd„de ƒZ Gdd„de ƒZGdd„de ƒZGdd„deeƒZdS)é)ÚIterableÚAny)ÚTrialé)ÚPipeline)ÚMappingc@seZdZdZdS)Ú ParameterzBase hyper-parameterN)Ú__name__Ú __module__Ú __qualname__Ú__doc__©r r úO/home/ubuntu/.local/lib/python3.10/site-packages/pyannote/pipeline/parameter.pyr&srcó8eZdZdZdef‡fdd„ Zdedefdd„Z‡Z S) Ú Categoricalz­Categorical hyper-parameter The value is sampled from `choices`. Parameters ---------- choices : iterable Candidates of hyper-parameter value. Úchoicescstƒ ¡t|ƒ|_dS©N)ÚsuperÚ__init__Úlistr)Úselfr©Ú __class__r rr7s zCategorical.__init__ÚnameÚtrialcCs| ||j¡Sr)Úsuggest_categoricalr©rrrr r rÚ__call__;szCategorical.__call__) r r r r rrÚstrrrÚ __classcell__r r rrr,ó rcs@eZdZdZdededef‡fdd„ Zdedefd d „Z‡Z S) ÚDiscreteUniforma¿Discrete uniform hyper-parameter The value is sampled from the range [low, high], and the step of discretization is `q`. Parameters ---------- low : `float` Lower endpoint of the range of suggested values. `low` is included in the range. high : `float` Upper endpoint of the range of suggested values. `high` is included in the range. q : `float` A step of discretization. ÚlowÚhighÚqcs,tƒ ¡t|ƒ|_t|ƒ|_t|ƒ|_dSr)rrÚfloatr"r#r$)rr"r#r$rr rrQs   zDiscreteUniform.__init__rrcCs| ||j|j|j¡Sr)Úsuggest_discrete_uniformr"r#r$rr r rrWszDiscreteUniform.__call__© r r r r r%rrrrrr r rrr!?sr!có<eZdZdZdedef‡fdd„ Zdedefdd „Z‡Z S) ÚIntegera[Integer hyper-parameter The value is sampled from the integers in [low, high]. Parameters ---------- low : `int` Lower endpoint of the range of suggested values. `low` is included in the range. high : `int` Upper endpoint of the range of suggested values. `high` is included in the range. r"r#có"tƒ ¡t|ƒ|_t|ƒ|_dSr)rrÚintr"r#©rr"r#rr rrjó  zInteger.__init__rrcCó| ||j|j¡Sr)Ú suggest_intr"r#rr r rroózInteger.__call__) r r r r r+rrrrrr r rrr)[ór)cr() Ú LogUniformaqLog-uniform hyper-parameter The value is sampled from the range [low, high) in the log domain. Parameters ---------- low : `float` Lower endpoint of the range of suggested values. `low` is included in the range. high : `float` Upper endpoint of the range of suggested values. `high` is excluded from the range. r"r#cr*r©rrr%r"r#r,rr rr‚r-zLogUniform.__init__rrcCr.r)Úsuggest_loguniformr"r#rr r rr‡r0zLogUniform.__call__r'r r rrr2sr1r2cr() ÚUniformapUniform hyper-parameter The value is sampled from the range [low, high) in the linear domain. Parameters ---------- low : `float` Lower endpoint of the range of suggested values. `low` is included in the range. high : `float` Upper endpoint of the range of suggested values. `high` is excluded from the range. r"r#cr*rr3r,rr rršr-zUniform.__init__rrcCr.r)Úsuggest_uniformr"r#rr r rrŸr0zUniform.__call__r'r r rrr5‹r1r5cr) ÚFrozenz|Frozen hyper-parameter The value is fixed a priori Parameters ---------- value : Fixed value. Úvaluecstƒ ¡||_dSr)rrr8)rr8rr rr®s  zFrozen.__init__rrcCs|jSr)r8rr r rr²szFrozen.__call__) r r r r rrrrrrr r rrr7£r r7cs8eZdZdZ‡fdd„Zdd„Zdd„Zdd „Z‡ZS) Ú ParamDictzèDict-like structured hyper-parameter Usage ----- >>> params = ParamDict(param1=Uniform(0.0, 1.0), param2=Uniform(-1.0, 1.0)) >>> params = ParamDict(**{"param1": Uniform(0.0, 1.0), "param2": Uniform(-1.0, 1.0)}) c s2tƒ ¡||_| ¡D] \}}t|||ƒq dSr)rrÚ_ParamDict__paramsÚitemsÚsetattr)rÚparamsÚ param_nameÚ param_valuerr rr¿s ÿzParamDict.__init__cCó t|jƒSr)Úlenr:©rr r rÚ__len__Åó zParamDict.__len__cCr@r)Úiterr:rBr r rÚ__iter__ÈrDzParamDict.__iter__cCs t||ƒSr)Úgetattr)rr>r r rÚ __getitem__ËrDzParamDict.__getitem__) r r r r rrCrFrHrr r rrr9¶s  r9N)ÚtypingrrÚ optuna.trialrÚpipelinerÚcollections.abcrrrr!r)r2r5r7r9r r r rÚs