From e348aa825d805c6158d1dd8c0ca1f253f40a8693 Mon Sep 17 00:00:00 2001 From: WongLaw Date: Wed, 26 Oct 2022 06:12:27 +0000 Subject: [PATCH] Added Rhythm Prediction, test=tts --- paddlespeech/text/exps/ernie_linear/test.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/paddlespeech/text/exps/ernie_linear/test.py b/paddlespeech/text/exps/ernie_linear/test.py index 68c6c2cb..aa172cc6 100644 --- a/paddlespeech/text/exps/ernie_linear/test.py +++ b/paddlespeech/text/exps/ernie_linear/test.py @@ -19,13 +19,14 @@ import pandas as pd import yaml from paddle import nn from paddle.io import DataLoader +from sklearn.metrics import classification_report +from sklearn.metrics import precision_recall_fscore_support +from yacs.config import CfgNode + from paddlespeech.t2s.utils import str2bool from paddlespeech.text.models.ernie_linear import ErnieLinear from paddlespeech.text.models.ernie_linear import PuncDataset from paddlespeech.text.models.ernie_linear import PuncDatasetFromErnieTokenizer -from sklearn.metrics import classification_report -from sklearn.metrics import precision_recall_fscore_support -from yacs.config import CfgNode DefinedClassifier = { 'ErnieLinear': ErnieLinear,