コード例 #1
0
ファイル: RulePrunerTest.java プロジェクト: kristiankode/ID3
  @Test
  public void prune_givenValidationDataSameAsTrainingData_expect100Percent()
      throws FileNotFoundException, UnsupportedEncodingException {

    Model model = data.getMushroomModel();
    List<Sample> trainingSet = data.getAllMushroomSamples();
    List<Prediction> originalPrediction = predictor.predict(model, trainingSet);
    Double originalAccuracy = accuracy.evaluate(originalPrediction, model.getTargetAttribute());

    List<Rule> prunedRules = pruner.pruneRepeatedly(model, trainingSet);
    List<Prediction> prunedPrediction = predictor.predict(prunedRules, trainingSet);
    Double prunedAccuracy = accuracy.evaluate(prunedPrediction, model.getTargetAttribute());

    log.debug(
        "Accuracy before pruning = {}, Accuracy after pruning = {}",
        originalAccuracy,
        prunedAccuracy);

    assertThat(originalAccuracy, is(prunedAccuracy));
  }