qolmat.benchmark.metrics.dist_wasserstein¶
- qolmat.benchmark.metrics.dist_wasserstein(df1: DataFrame, df2: DataFrame, df_mask: DataFrame, method: str = 'columnwise') Series[source]¶
Compute the Wasserstein distances between columns of 2 dataframes.
Wasserstein distance can only be computed columnwise.
- Parameters
- df1pd.DataFrame
True dataframe
- df2pd.DataFrame
Predicted dataframe
- df_maskpd.DataFrame
Elements of the dataframes to compute on
- methodstr, optional
columnwise or not
- Returns
- pd.Series
wasserstein distances