Browse Source

Fixed Prediction Score Bug

master
hwy893747147 3 years ago
parent
commit
4da292719d
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      tods/sk_interface/base.py

+ 2
- 2
tods/sk_interface/base.py View File

@@ -61,7 +61,7 @@ class BaseSKI:
raise AttributeError('type object ' + self.__class__.__name__ + ' has no attribute \'predict\'')

data = self._sys_data_check(data)
output_data = self._forward(data, 'produce')
output_data = self._forward(data, '_produce')

return output_data
@@ -71,7 +71,7 @@ class BaseSKI:
raise AttributeError('type object ' + self.__class__.__name__ + ' has no attribute \'predict_score\'')

data = self._sys_data_check(data)
output_data = self._forward(data, 'produce_score')
output_data = self._forward(data, '_produce_score')

return output_data



Loading…
Cancel
Save