Update prosody_embedding_pipeline.py
Browse files
prosody_embedding_pipeline.py
CHANGED
|
@@ -159,7 +159,7 @@ class ProsodyEmbeddingPipeline(Pipeline):
|
|
| 159 |
diff_count = np.sum(diff_points)
|
| 160 |
total_points = len(input_f0_np)
|
| 161 |
|
| 162 |
-
|
| 163 |
|
| 164 |
|
| 165 |
|
|
@@ -180,7 +180,7 @@ class ProsodyEmbeddingPipeline(Pipeline):
|
|
| 180 |
outputs['intensity_recon'] = output_intensity_np
|
| 181 |
# Add metrics to outputs
|
| 182 |
outputs['metrics'] = {
|
| 183 |
-
'
|
| 184 |
'intensity_rmse': float(intensity_rmse)
|
| 185 |
}
|
| 186 |
|
|
|
|
| 159 |
diff_count = np.sum(diff_points)
|
| 160 |
total_points = len(input_f0_np)
|
| 161 |
|
| 162 |
+
f0_large_diff_percent = (diff_count / total_points) * 100
|
| 163 |
|
| 164 |
|
| 165 |
|
|
|
|
| 180 |
outputs['intensity_recon'] = output_intensity_np
|
| 181 |
# Add metrics to outputs
|
| 182 |
outputs['metrics'] = {
|
| 183 |
+
'f0_large_diff_percent': f0_large_diff_percent.item(),
|
| 184 |
'intensity_rmse': float(intensity_rmse)
|
| 185 |
}
|
| 186 |
|