4bdc700248d5a99e7bebe03e767c0befa6be7803,finetune/target_models/comparison_regressor.py,ComparisonRegressor,predict,#ComparisonRegressor#Any#Any#,59

Before Change

        :param pairs: Array of text, shape [batch, 2]
        :returns: list of floats, shape [batch]
        
        return super().predict(pairs=pairs, context=context, **kwargs)
    
    def _predict(self, pairs, context=None, **kwargs):
        
        Produces a floating point prediction determined by the fine-tuned model.

After Change

        :param pairs: Array of text, shape [batch, 2]
        :returns: list of floats, shape [batch]
        
        zipped_data = self.input_pipeline.zip_list_to_dict(X=pairs, context=context)
        raw_preds = self._inference(zipped_data, predict_keys=[PredictMode.NORMAL], context=context, **kwargs)
        return self.input_pipeline.label_encoder.inverse_transform(np.asarray(raw_preds))
    
    def predict_proba(self, pairs, context=None):
        
        Not implemented in regression task.

Instances