help='pure training')
parser.add_argument('--test', type=bool, default=False, help='test_phase')
parser.add_argument('--novel_path', type=str, help='training path')
#parser.add_argument('--train_path',type=str,default='task2-dataset/base',help='testing path')
parser.add_argument('--test_path', type=str, help='testing path')
parser.add_argument('--ensemble',
type=bool,
default=False,
help='ensemble?')
parser.add_argument('--way', type=bool, default=False, help='ensemble?')
parser.add_argument('--outfile', type=str, help='outfile_csv')
args = parser.parse_args()
print(args)
import Task2
model = Task2.Task2(args)
#print(args.train)
if args.train:
model.train()
elif args.test:
model.test()
else:
model.ensemble()
'''
import Task2_1
model = Task2_1.Task2(args)
if args.way:
model.train()
import Task1 import Task2 import Task3 # regression task task_1 = Task1.Task1() task_1.model_1_run() task_1.model_2_run() # multi-category task task_2 = Task2.Task2() task_2.model_1_run() task_2.model_2_run() # multi-label task task_3 = Task3.Task3() task_3.model_1_run() task_3.model_2_run()
