Python运行时把训练和测试分开运行（vscode）

一、添加代码

在main.py文件中，首先在开头添加如下代码：

assert len(sys.argv) == 2
state = sys.argv[1]
assert state in ['train', 'predict']

在训练代码前添加if语句，并把if中的内容集体后移（vscode可以用Ctrl+]键进行集体后移）

if state == 'train':

同样在测试代码前做相同操作：

elif state == 'predict':

二、终端操作运行程序

在终端运行如下代码即为只运行train部分：

python main.py train

在终端运行如下代码即为只运行predict部分：

python main.py predict