feat: minor calc update

biosimulator_processes/processes/comparator_process.py

@@ -6,8 +6,22 @@
 from biosimulator_processes.data_model.compare_data_model import ComparisonResults, SimulatorResult, IntervalResult
 
 
-def mean_squared_error(true_values: np.ndarray, predicted_values: np.ndarray):
- return np.mean((predicted_values - true_values) ** 2)
+def mean_squared_error(data: Union[List[float, np.ndarray]]) -> np.ndarray[float]:
+ """Takes in an array of data which represents the results of a composition for a species/param
+ at a given interval, averages the values, and then scores each item in the data array with an MSE value
+ calculated against the average, returning the MSE scores as an array.
+
+ Args:
+ data:`Union[List[float, np.ndarray[float]]]`: An array of data which represents the results
+ of a composition for a species/param.
+
+ Returns:
+ `np.ndarray[float]`: An array of MSE scores for each item in `data`.
+ """
+ if isinstance(data, list):
+ data = np.array(data)
+ data_average = np.mean(data)
+ return (data - data_average) ** 2
 
 
 class CompareResults(Step):