Examples with AkkaActorInputMessage ml.shifu.shifu.message.AkkaActorInputMessage used on opensource projects

Example 16 with AkkaActorInputMessage

use of ml.shifu.shifu.message.AkkaActorInputMessage in project shifu by ShifuML.

the class EvalModelActorTest method testActor.

// @Test
public void testActor() throws IOException, InterruptedException {
    Environment.setProperty(Environment.SHIFU_HOME, ".");
    File tmpModels = new File("models");
    File tmpCommon = new File("common");
    File models = new File("src/test/resources/example/cancer-judgement/ModelStore/ModelSet1/models");
    FileUtils.copyDirectory(models, tmpModels);
    File tmpEvalA = new File("evals");
    FileUtils.forceMkdir(tmpEvalA);
    ActorRef modelEvalRef = actorSystem.actorOf(new Props(new UntypedActorFactory() {

        private static final long serialVersionUID = -1437127862571741369L;

        public UntypedActor create() {
            return new EvalModelActor(modelConfig, columnConfigList, new AkkaExecStatus(true), evalConfig);
        }
    }), "model-evaluator");
    List<Scanner> scanners = ShifuFileUtils.getDataScanners("src/test/resources/example/cancer-judgement/DataStore/EvalSet1", SourceType.LOCAL);
    modelEvalRef.tell(new AkkaActorInputMessage(scanners), modelEvalRef);
    while (!modelEvalRef.isTerminated()) {
        Thread.sleep(5000);
    }
    File outputFile = new File("evals/EvalA/EvalScore");
    Assert.assertTrue(outputFile.exists());
    FileUtils.deleteDirectory(tmpModels);
    FileUtils.deleteDirectory(tmpCommon);
    FileUtils.deleteDirectory(tmpEvalA);
}

Example 17 with AkkaActorInputMessage

use of ml.shifu.shifu.message.AkkaActorInputMessage in project shifu by ShifuML.

the class PostTrainActorTest method testActor.

// @Test
public void testActor() throws IOException, InterruptedException {
    File tmpModels = new File("models");
    File tmpCommon = new File("common");
    File models = new File("src/test/resources/example/cancer-judgement/ModelStore/ModelSet1/models");
    FileUtils.copyDirectory(models, tmpModels);
    ActorRef postTrainRef = actorSystem.actorOf(new Props(new UntypedActorFactory() {

        private static final long serialVersionUID = 6777309320338075269L;

        public UntypedActor create() {
            return new PostTrainActor(modelConfig, columnConfigList, new AkkaExecStatus(true));
        }
    }), "post-trainer");
    List<Scanner> scanners = ShifuFileUtils.getDataScanners("src/test/resources/example/cancer-judgement/DataStore/DataSet1", SourceType.LOCAL);
    postTrainRef.tell(new AkkaActorInputMessage(scanners), postTrainRef);
    while (!postTrainRef.isTerminated()) {
        Thread.sleep(5000);
    }
    File file = new File("./ColumnConfig.json");
    Assert.assertTrue(file.exists());
    FileUtils.deleteQuietly(file);
    FileUtils.deleteDirectory(tmpModels);
    FileUtils.deleteDirectory(tmpCommon);
}