Open simo-an opened 2 years ago
I just used the following hard code to evaluate (in eval.py line 258):
roc_auc = eval_model(model_name, 'bottle', save_plots=args.save_plots, device=device, head_layer=args.head_layer, density=density())
Thanks!
You hard coded bottle
as the second argument. So I suspect you get the AUC of that class.
Instead of hardcoding you could also use the eval script:
python eval.py --head_layer 2 --type wood
I train a model of wood using the command:
Then in the evaluate process, is use the model to evaluate the type of bottle , I got a surprisingly excellent result:
Is there any problem here?
I just used the following hard code to evaluate (in eval.py line 258):
Thanks!