small format changes

Ziruiwang409 · Dec 13, 2023 · ad93049 · ad93049
1 parent 589dcd9
commit ad93049
Show file tree

Hide file tree

Showing 6 changed files with 5 additions and 60 deletions.
diff --git a/metric/WER_CER_metric.py b/metric/WER_CER_metric.py
diff --git a/WERandCERmetrics.py → metric/WERandCERmetrics.py b/WERandCERmetrics.py → metric/WERandCERmetrics.py
diff --git a/evaluation.py → metric/evaluation.py b/evaluation.py → metric/evaluation.py
diff --git a/test.py b/test.py
@@ -11,7 +11,7 @@
 from dataset import dataset
 
 # evaluation
-from metric.WER_CER_metric import CalculateErrorRate
+from metric.WERandCERmetrics import CalculateErrorRate
 from misc import idx2text, ctc_decoder, plot_error_curves_comparison
 
 def parse_args():

diff --git a/train.py b/train.py
@@ -6,15 +6,12 @@
 import numpy as np
 import argparse
 
-# statistics & visualization
-import matplotlib.pyplot as plt
-
 # model & dataset
 from model.densenet_3d import DenseNet3D
 from dataset import dataset
 
 # evaluation
-from metric.WER_CER_metric import CalculateErrorRate
+from metric.WERandCERmetrics import CalculateErrorRate
 from misc import idx2text, ctc_decoder, plot_error_curves_comparison
 
 def parse_args():

diff --git a/train_script.ipynb b/train_script.ipynb