Skip to content

Commit

Permalink
Fix sample_count
Browse files Browse the repository at this point in the history
Co-Authored-By: gilbertoolimpio <[email protected]>
  • Loading branch information
jacobmontiel and giba0 authored Nov 8, 2018
1 parent 4e81758 commit 522176a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/skmultiflow/metrics/measure_collection.py
Original file line number Diff line number Diff line change
Expand Up @@ -948,7 +948,7 @@ def add_result(self, y_true, y_pred, weight=1.0):

self.total_square_error += (y_true - y_pred) * (y_true - y_pred)
self.average_error += np.absolute(y_true - y_pred)
self.sample_count += weight
self.sample_count += 1

def get_mean_square_error(self):
""" Computes the mean square error.
Expand Down

0 comments on commit 522176a

Please sign in to comment.