cceae1fee29ec0269616e7c2364f1714a518b2c3,python/src/vmaf/core/train_test_model.py,TrainTestModel,from_file,#Any#Any#Any#Any#,338

Before Change

        train_test_model.param_dict = info_loaded["param_dict"]
        train_test_model.model_dict = info_loaded["model_dict"]

        if issubclass(model_class, LibsvmNusvrTrainTestModel):
            // == special handling of libsvmnusvr: load .model differently ==
            model = svmutil.svm_load_model(filename + ".model")
            train_test_model.model_dict["model"] = model

        return train_test_model

    def _preproc_train(self, xys):
        self.model_type = self.TYPE

After Change

    @classmethod
    def from_file(cls, filename, logger=None, optional_dict2=None):
        assert os.path.exists(filename), "File name {} does not exist.".format(filename)
        with open(filename, "rb") as file:
            info_loaded = pickle.load(file)
        model_type = info_loaded["model_dict"]["model_type"]
        model_class = TrainTestModel.find_subclass(model_type)

Instances