예제 #1
0
파일: executor.py 프로젝트: Netflix/vmaf
    def __init__(self,
                 assets,
                 logger,
                 fifo_mode=True,
                 delete_workdir=True,
                 result_store=None,
                 optional_dict=None,
                 optional_dict2=None,
                 ):
        """
        Use optional_dict for parameters that would impact result (e.g. model,
        patch size), and use optional_dict2 for parameters that would NOT
        impact result (e.g. path to data cache file).
        """

        TypeVersionEnabled.__init__(self)

        self.assets = assets
        self.logger = logger
        self.fifo_mode = fifo_mode
        self.delete_workdir = delete_workdir
        self.results = []
        self.result_store = result_store
        self.optional_dict = optional_dict
        self.optional_dict2 = optional_dict2

        self._assert_args()
        self._assert_assets()
예제 #2
0
    def __init__(
        self,
        assets,
        logger,
        fifo_mode=True,
        delete_workdir=True,
        result_store=None,
        optional_dict=None,
        optional_dict2=None,
    ):
        """
        Use optional_dict for parameters that would impact result (e.g. model,
        patch size), and use optional_dict2 for parameters that would NOT
        impact result (e.g. path to data cache file).
        """

        TypeVersionEnabled.__init__(self)

        self.assets = assets
        self.logger = logger
        self.fifo_mode = fifo_mode
        self.delete_workdir = delete_workdir
        self.results = []
        self.result_store = result_store
        self.optional_dict = optional_dict
        self.optional_dict2 = optional_dict2

        self._assert_class()
        self._assert_args()
        self._assert_assets()
예제 #3
0
    def __init__(self, param_dict, logger=None):
        """
        :param param_dict: contains input parameters
        :param logger:
        :return:
        """
        TypeVersionEnabled.__init__(self)

        self.param_dict = param_dict
        self.logger = logger

        self.model_dict = {}
예제 #4
0
    def __init__(self, param_dict, logger=None):
        """
        :param param_dict: contains input parameters
        :param logger:
        :return:
        """
        TypeVersionEnabled.__init__(self)

        self.param_dict = param_dict
        self.logger = logger

        self.model_dict = {}
예제 #5
0
 def __init__(self, groundtruths, predictions):
     """
     Performance metric on quality metrics
     :param groundtruths: either list of real numbers (aggregate scores like
     MOS or DMOS or MLE), or list of list of real numbers (list of raw scores)
     :param predictions: list of real numbers
     :return:
     """
     TypeVersionEnabled.__init__(self)
     self.groundtruths = groundtruths
     self.predictions = predictions
     self._assert_args()
예제 #6
0
파일: perf_metric.py 프로젝트: Netflix/vmaf
 def __init__(self, groundtruths, predictions):
     """
     Performance metric on quality metrics
     :param groundtruths: either list of real numbers (aggregate scores like
     MOS or DMOS or MLE), or list of list of real numbers (list of raw scores)
     :param predictions: list of real numbers
     :return:
     """
     TypeVersionEnabled.__init__(self)
     self.groundtruths = groundtruths
     self.predictions = predictions
     self._assert_args()
예제 #7
0
    def __init__(self, param_dict, logger=None, optional_dict2=None):
        '''
        Put in optional_dict2 optionals that would not impact result, e.g.
        path to checkpoint file directories, or h5py file
        '''
        TypeVersionEnabled.__init__(self)
        self.param_dict = param_dict
        self.logger = logger
        self.optional_dict2 = optional_dict2

        self.model_dict = {}

        self._assert_args()
예제 #8
0
    def __init__(self, param_dict, logger=None, optional_dict2=None):
        '''
        Put in optional_dict2 optionals that would not impact result, e.g.
        path to checkpoint file directories, or h5py file
        '''
        TypeVersionEnabled.__init__(self)
        self.param_dict = param_dict
        self.logger = logger
        self.optional_dict2 = optional_dict2

        self.model_dict = {}

        self._assert_args()
예제 #9
0
    def __init__(self,
                 assets,
                 logger,
                 fifo_mode=True,
                 delete_workdir=True,
                 result_store=None,
                 optional_dict=None,
                 ):

        TypeVersionEnabled.__init__(self)

        self.assets = assets
        self.logger = logger
        self.fifo_mode = fifo_mode
        self.delete_workdir = delete_workdir
        self.results = []
        self.result_store = result_store
        self.optional_dict = optional_dict

        self._assert_assets()
예제 #10
0
    def executor_id(self):
        executor_id_ = TypeVersionEnabled.get_type_version_string(self)

        if self.optional_dict is not None and len(self.optional_dict) > 0:
            # include optional_dict info in executor_id for result store,
            # as parameters in optional_dict will impact result
            executor_id_ += '_{}'.format('_'.join(
                map(lambda k: '{k}_{v}'.format(k=k, v=self.optional_dict[k]),
                    sorted(self.optional_dict.keys()))))

        return executor_id_
예제 #11
0
파일: executor.py 프로젝트: Netflix/vmaf
    def executor_id(self):
        executor_id_ = TypeVersionEnabled.get_type_version_string(self)

        if self.optional_dict is not None:
            # include optional_dict info in executor_id for result store,
            # as parameters in optional_dict will impact result
            executor_id_ += '_{}'.format(
                '_'.join(
                    map(lambda k: '{k}_{v}'.format(k=k,v=self.optional_dict[k]),
                        sorted(self.optional_dict.keys()))
                )
            )

        return executor_id_
예제 #12
0
 def model_id(self):
     return TypeVersionEnabled.get_type_version_string(self)
예제 #13
0
 def executor_id(self):
     return TypeVersionEnabled.get_type_version_string(self)
예제 #14
0
 def __init__(self, dataset_reader):
     TypeVersionEnabled.__init__(self)
     self.dataset_reader = dataset_reader
예제 #15
0
 def model_id(self):
     return TypeVersionEnabled.get_type_version_string(self)
예제 #16
0
 def __init__(self, dataset_reader):
     TypeVersionEnabled.__init__(self)
     self.dataset_reader = dataset_reader