a [=ic+@sdZddlZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd l m Zdd lmZdd lmZdd lmZddZddZGdddejZGdddejZdddZdS)z?Contains functions for evaluation and summarization of metrics.N)dtypes)ops) array_ops)init_ops)math_ops) state_ops)variable_scope) tf_logging)basic_session_run_hooks)monitored_session)session_run_hookcCspt}|tjj}t|dkr*|dSt|dkr@tdn,tjdgt j t dtjj tjjgd}|SdS)zGets or creates the eval step `Tensor`. Returns: A `Tensor` representing a counter for the evaluation step. Raises: ValueError: If multiple `Tensors` have been added to the `tf.GraphKeys.EVAL_STEP` collection. rz0Multiple tensors added to tf.GraphKeys.EVAL_STEP eval_stepF)shapedtypeZ initializerZ trainable collectionsN)rZget_default_graphZget_collectionZ GraphKeysZ EVAL_STEPlen ValueErrorrZ get_variablerint64rZzeros_initializerZLOCAL_VARIABLES)graphZ eval_stepscounterrv/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/tensorflow/python/training/evaluation.py_get_or_create_eval_step!s    rcCsTt|trt|}t| tt WdS1sF0YdS)aGets the eval step `Tensor` value after running `update_ops`. Args: update_ops: A list of `Tensors` or a dictionary of names to `Tensors`, which are run before reading the eval step value. Returns: A `Tensor` representing the value for the evaluation step. N) isinstancedictlistvaluesrZcontrol_dependenciesridentityrZ read_value)Z update_opsrrr_get_latest_eval_step_value<s   rc@sBeZdZdZdddZddZddZd d Zd d Zd dZ dS)_MultiStepStopAfterNEvalsHookGRun hook used by the evaluation routines to run the `eval_ops` N times.r cCs||_d|_||_dS)zConstructs the run hook. Args: num_evals: The number of evaluations to run for. if set to None, will iterate the dataset until all inputs are exhausted. steps_per_run: Number of steps executed per run call. N) _num_evals_evals_completed_steps_per_run_initial_value)self num_evalsZ steps_per_runrrr__init__Psz&_MultiStepStopAfterNEvalsHook.__init__cCs ||_dSNr#r%Zupdated_eval_steprrr_set_evals_completed_tensor\sz9_MultiStepStopAfterNEvalsHook._set_evals_completed_tensorcCst|_dSr()r $get_or_create_steps_per_run_variable_steps_per_run_variable)r%rrrbegin_sz#_MultiStepStopAfterNEvalsHook.begincCs4|jdur|j}nt|j|j}|jj||ddS)Nsession)r"r$minr-load)r%r0Zcoordstepsrrrafter_create_sessioncs z2_MultiStepStopAfterNEvalsHook.after_create_sessioncCstd|jiSNevals_completedr ZSessionRunArgsr#r% run_contextrrr before_runksz(_MultiStepStopAfterNEvalsHook.before_runcCs|jd}|jdur|j}nt|j||j}|jj||jd|jdurXtd|ntd||j|jdur||jkr| dS)Nr6r/Evaluation [%d]Evaluation [%d/%d]) resultsr"r$r1r-r2r0logginginfo request_stop)r%r9 run_valuesr6r3rrr after_runos    z'_MultiStepStopAfterNEvalsHook.after_runN)r ) __name__ __module__ __qualname____doc__r'r+r.r4r:rBrrrrr Ms r c@s2eZdZdZd ddZddZddZd d Zd S) _StopAfterNEvalsHookr!TcCs:||_d|_||_|dus"|dkr&dn t|d|_dS)zConstructs the run hook. Args: num_evals: The number of evaluations to run for. if set to None, will iterate the dataset until all inputs are exhausted. log_progress: Whether to log evaluation progress, defaults to True. Nr g$@)r"r# _log_progressmathfloor_log_frequency)r%r&Z log_progressrrrr's  z_StopAfterNEvalsHook.__init__cCs ||_dSr(r)r*rrrr+sz0_StopAfterNEvalsHook._set_evals_completed_tensorcCstd|jiSr5r7r8rrrr:sz_StopAfterNEvalsHook.before_runcCsp|jd}|jrP|jdur(td|n(||jdks@|j|krPtd||j|jdurl||jkrl|dS)Nr6r;rr<)r=rIr"r>r?rLr@)r%r9rAr6rrrrBs  z_StopAfterNEvalsHook.after_runN)T)rCrDrErFr'r+r:rBrrrrrGs  rGc Cst} t|pg}|durtdd|DrRt} tj| tj| | j ddd} ntj| ddd} t |t rv| |d<n&t |t tfrt|| g}n|| g}t |} |D]} t | ttfr| | qtd td tt}tj||||d }t||}||tj||d 4}|durD|sD|||q*Wdn1sZ0Ytd t|tdtdt|jS)ay Evaluates the model at the given checkpoint path. During a single evaluation, the `eval_ops` is run until the session is interrupted or requested to finish. This is typically requested via a `tf.contrib.training.StopAfterNEvalsHook` which results in `eval_ops` running the requested number of times. Optionally, a user can pass in `final_ops`, a single `Tensor`, a list of `Tensors` or a dictionary from names to `Tensors`. The `final_ops` is evaluated a single time after `eval_ops` has finished running and the fetched values of `final_ops` are returned. If `final_ops` is left as `None`, then `None` is returned. One may also consider using a `tf.contrib.training.SummaryAtEndHook` to record summaries after the `eval_ops` have run. If `eval_ops` is `None`, the summaries run immediately after the model checkpoint has been restored. Note that `evaluate_once` creates a local variable used to track the number of evaluations run via `tf.contrib.training.get_or_create_eval_step`. Consequently, if a custom local init op is provided via a `scaffold`, the caller should ensure that the local init op also initializes the eval step. Args: checkpoint_path: The path to a checkpoint to use for evaluation. master: The BNS address of the TensorFlow master. scaffold: An tf.compat.v1.train.Scaffold instance for initializing variables and restoring variables. Note that `scaffold.init_fn` is used by the function to restore the checkpoint. If you supply a custom init_fn, then it must also take care of restoring the model from its checkpoint. eval_ops: A single `Tensor`, a list of `Tensors` or a dictionary of names to `Tensors`, which is run until the session is requested to stop, commonly done by a `tf.contrib.training.StopAfterNEvalsHook`. feed_dict: The feed dictionary to use when executing the `eval_ops`. final_ops: A single `Tensor`, a list of `Tensors` or a dictionary of names to `Tensors`. final_ops_feed_dict: A feed dictionary to use when evaluating `final_ops`. hooks: List of `tf.estimator.SessionRunHook` callbacks which are run inside the evaluation loop. config: An instance of `tf.compat.v1.ConfigProto` that will be used to configure the `Session`. If left as `None`, the default will be used. Returns: The fetched values of `final_ops` or `None` if `final_ops` is `None`. Ncss|]}t|tVqdSr()rr ).0hrrr z!_evaluate_once..)rT)Z use_lockingr update_eval_stepzStarting evaluation at z%Y-%m-%dT%H:%M:%S)scaffoldZcheckpoint_filename_with_pathmasterconfig)session_creatorhookszInference Time : {:0.5f}szFinished evaluation at z%Y-%m-%d-%H:%M:%S)rranyr r,rZ assign_addrcastrrrtuplerrGr r+r>r?timestrftime localtimer ZChiefSessionCreatorZ FinalOpsHookappendZMonitoredSessionZ should_stoprunformatZfinal_ops_values)Zcheckpoint_pathrTrSZeval_opsZ feed_dictZ final_opsZfinal_ops_feed_dictrWrUrZsteps_per_run_variablerRZeval_step_valuerOstartrVZfinal_ops_hookr0rrr_evaluate_onces^5       0rb)rMNNNNNNN)rFrJr[Ztensorflow.python.frameworkrrZtensorflow.python.opsrrrrrZtensorflow.python.platformr r>Ztensorflow.python.trainingr r r rrZSessionRunHookr rGrbrrrrs4           4(