0
|
1 package de.mpiwg.anteater.results.impl;
|
|
2
|
|
3 import java.util.List;
|
|
4
|
|
5 import de.mpiwg.anteater.AnteaterConfiguration;
|
|
6 import de.mpiwg.anteater.ml.MLController;
|
|
7 import de.mpiwg.anteater.results.ApplicantResult;
|
|
8 import de.mpiwg.anteater.results.IResultFinder;
|
|
9 import de.mpiwg.anteater.results.LocationResult;
|
|
10 import de.mpiwg.anteater.results.SpeciesScientificResult;
|
|
11 import de.mpiwg.anteater.text.TextInformation;
|
|
12
|
|
13 public class LocationResultFinder implements IResultFinder<LocationResult> {
|
|
14
|
|
15 private AnteaterConfiguration configuration;
|
|
16 private List<SpeciesScientificResult> predictedSpecies;
|
|
17 private List<ApplicantResult> predictedApplicants;
|
|
18
|
|
19 public LocationResultFinder(AnteaterConfiguration configuration, List<SpeciesScientificResult> predictedSpecies, List<ApplicantResult> predictedApplicants) {
|
|
20 this.configuration = configuration;
|
|
21 this.predictedApplicants = predictedApplicants;
|
|
22 this.predictedSpecies = predictedSpecies;
|
|
23 }
|
|
24
|
|
25 @Override
|
|
26 public List<LocationResult> getResults(List<TextInformation> infos) {
|
|
27 MLController mlController = new MLController(configuration);
|
|
28 return mlController.runLocationMLComponent(infos, predictedSpecies, predictedApplicants);
|
|
29 }
|
|
30
|
|
31 }
|