MycroftAI
/
mycroft-precise
mirror of https://github.com/MycroftAI/mycroft-precise.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix loss bias in trainer.py

pull/23/merge
Matthew Scholefield 2018-08-10 00:02:39 -05:00
parent 63b4592675
commit 02fa053653
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
precise/scripts/train.py
View File

@ -19,7 +19,6 @@ from os.path import splitext, isfile
from prettyparse import add_to_parser
from typing import Any, Tuple
from precise.functions import set_loss_bias
from precise.model import create_model, ModelParams
from precise.params import inject_params, save_params
from precise.train_data import TrainData
@ -82,8 +81,8 @@ class Trainer:
save_params(args.model)
self.train, self.test = self.load_data(self.args)
set_loss_bias(1.0 - args.sensitivity)
params = ModelParams(skip_acc=args.no_validation, extra_metrics=args.extra_metrics)
params = ModelParams(skip_acc=args.no_validation, extra_metrics=args.extra_metrics,
loss_bias=1.0 - args.sensitivity)
self.model = create_model(args.model, params)
self.model.summary()