Matrix _mean method corrected

gyrdym · Oct 9, 2019 · 4f22b59 · 4f22b59
1 parent 4adecbb
commit 4f22b59
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/lib/src/matrix/base_matrix.dart b/lib/src/matrix/base_matrix.dart
@@ -169,22 +169,21 @@ abstract class BaseMatrix with
 
     switch (axis) {
       case Axis.columns:
-        return _meansCache[axis] ??= _mean(columns, rowsNum);
+        return _meansCache[axis] ??= _mean(columns);
 
       case Axis.rows:
-        return _meansCache[axis] ??= _mean(rows, columnsNum);
+        return _meansCache[axis] ??= _mean(rows);
 
       default:
         throw UnimplementedError('Mean values calculation for axis $axis is not '
             'implemented yet');
     }
   }
 
-  Vector _mean(Iterable<Vector> vectors, int vectorsNum) =>
+  Vector _mean(Iterable<Vector> vectors) =>
       Vector.fromList(
         vectors.map((vector) => vector.mean()).toList(),
-        dtype: dtype,
-      );
+        dtype: dtype);
 
   @override
   Vector deviation([Axis axis = Axis.columns]) {