use of ml.shifu.shifu.message.TrainResultMessage in project shifu by ShifuML.
the class TrainModelWorker method handleMsg.
/* (non-Javadoc)
* @see akka.actor.UntypedActor#onReceive(java.lang.Object)
*/
@Override
public void handleMsg(Object message) throws IOException {
if (message instanceof TrainInstanceMessage) {
log.info("Received train data for model training");
TrainInstanceMessage msg = (TrainInstanceMessage) message;
msg.getTrainer().train();
nextActorRef.tell(new TrainResultMessage(), getSelf());
} else {
unhandled(message);
}
}
use of ml.shifu.shifu.message.TrainResultMessage in project shifu by ShifuML.
the class TrainDtModelActor method onReceive.
/* (non-Javadoc)
* @see akka.actor.UntypedActor#onReceive(java.lang.Object)
*/
@Override
public void onReceive(Object message) throws Exception {
if (message instanceof AkkaActorInputMessage) {
resultCnt = 0;
AkkaActorInputMessage msg = (AkkaActorInputMessage) message;
List<Scanner> scanners = msg.getScanners();
log.debug("Num of Scanners: " + scanners.size());
for (Scanner scanner : scanners) {
dataLoadRef.tell(new ScanStatsRawDataMessage(scanners.size(), scanner), getSelf());
}
} else if (message instanceof TrainResultMessage) {
resultCnt++;
if (resultCnt == trainerCnt) {
log.info("Received " + resultCnt + " finish message. Close System.");
Encog.getInstance().shutdown();
getContext().system().shutdown();
}
} else if (message instanceof ExceptionMessage) {
// since some children actors meet some exception, shutdown the system
ExceptionMessage msg = (ExceptionMessage) message;
getContext().system().shutdown();
// and wrapper the exception into Return status
addExceptionIntoCondition(msg.getException());
} else {
unhandled(message);
}
}
use of ml.shifu.shifu.message.TrainResultMessage in project shifu by ShifuML.
the class TrainModelActor method onReceive.
/* (non-Javadoc)
* @see akka.actor.UntypedActor#onReceive(java.lang.Object)
*/
@Override
public void onReceive(Object message) throws Exception {
if (message instanceof AkkaActorInputMessage) {
resultCnt = 0;
AkkaActorInputMessage msg = (AkkaActorInputMessage) message;
List<Scanner> scanners = msg.getScanners();
log.debug("Num of Scanners: " + scanners.size());
for (Scanner scanner : scanners) {
dataLoadRef.tell(new ScanTrainDataMessage(scanners.size(), isDryRun, scanner), getSelf());
}
} else if (message instanceof TrainResultMessage) {
resultCnt++;
if (resultCnt == trainerCnt) {
log.info("Received " + resultCnt + " finish message. Close System.");
Encog.getInstance().shutdown();
getContext().system().shutdown();
}
} else if (message instanceof ExceptionMessage) {
// since some children actors meet some exception, shutdown the system
ExceptionMessage msg = (ExceptionMessage) message;
getContext().system().shutdown();
// and wrapper the exception into Return status
addExceptionIntoCondition(msg.getException());
} else {
unhandled(message);
}
}
Aggregations