eu.openminted.uc.socialsciences.kb.preparation.eval.GoldData Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of omtd-uc-ss-kb-preparation Show documentation

The newest version!

package eu.openminted.uc.socialsciences.kb.preparation.eval;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.apache.log4j.Logger;

public class GoldData {

	private String datasetID;
	private Map> references;

	private static final Logger logger = Logger.getLogger(GoldData.class);

	public GoldData() {
		references = new HashMap<>();
	}

	public void setDatasetID(String datasetID) {
		this.datasetID = datasetID;
	}

	public String getDatasetID() {
		return datasetID;
	}

	public Map> getReferences() {
		return references;
	}

	public void addRef(String varRef, String refText, String paperRef) {
		if (null == varRef) {
			logger.warn("VarRef is null!");
		}
		List refs = references.getOrDefault(varRef, new ArrayList());
		refs.add(new Reference(refText, paperRef));
		references.put(varRef, refs);
	}

	@Override
	public String toString() {
		// for (String var : references.keySet()) {
		// List refs = references.get(var);
		//
		// }
		return String.format("Dataset ID: %s (%s)", datasetID, references.keySet().toString());
	}

}