amie.data.eval

Class PredictionsSampler

java.lang.Object

amie.data.eval.PredictionsSampler

public class PredictionsSampler
extends java.lang.Object

This class defines objects that take rules extracted from a training source and produce a sample of predictions: triples that are not in the training source.

Author:

lgalarra

Constructor Summary

Constructors

Constructor and Description
PredictionsSampler(KB dataset)
PredictionsSampler(KB dataset, int sampleSize)

Method Summary

Modifier and Type	Method and Description
java.lang.Object	generateBodyBindings(Rule rule) Given a rule, it produces a sample from the body bindings of the rule.
java.lang.Object	generateBodyPCABindings(Rule rule) Given a rule, it produces a sample from the body* bindings of the rule (the bindings that match the denominator of the PCA confidence expression)
java.util.Set<Triple<ByteString,ByteString,ByteString>>	generateBodyTriples(Rule rule, boolean PCAMode)
java.lang.Object	generatePredictions(Rule rule) Given a rule, it produces sample of predictions (triples that are beyond the database)
int	getSampleSize()
static void	main(java.lang.String[] args)
void	runMode1(java.util.Collection<Rule> rules) It takes a sample of unique predictions for the given rules by taking care of duplicated predictions.
void	runMode2(java.util.Collection<Rule> rules) It takes a sample of the predictions of every rule in the argument.
static java.util.Collection<Triple<ByteString,ByteString,ByteString>>	sample(java.util.Collection<Triple<ByteString,ByteString,ByteString>> samplingCandidates, int sampleSize) Given a collection of predictions, it extracts a random sample from it.
void	setNumberOfPredictions(int numberOfPredictions)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

PredictionsSampler

public PredictionsSampler(KB dataset)

PredictionsSampler

public PredictionsSampler(KB dataset,
                          int sampleSize)

Method Detail

getSampleSize

public int getSampleSize()

Returns:

the sampleSize

setNumberOfPredictions

public void setNumberOfPredictions(int numberOfPredictions)

Parameters:

sampleSize - the sampleSize to set

generateBodyTriples

public java.util.Set<Triple<ByteString,ByteString,ByteString>> generateBodyTriples(Rule rule,
                                                                                   boolean PCAMode)

generateBodyBindings

public java.lang.Object generateBodyBindings(Rule rule)

Given a rule, it produces a sample from the body bindings of the rule.

Parameters:

rule -

Returns:

generateBodyPCABindings

public java.lang.Object generateBodyPCABindings(Rule rule)

Given a rule, it produces a sample from the body* bindings of the rule (the bindings that match the denominator of the PCA confidence expression)

Parameters:

rule -

Returns:

generatePredictions

public java.lang.Object generatePredictions(Rule rule)

Given a rule, it produces sample of predictions (triples that are beyond the database)

Parameters:

rule -

Returns:

runMode1

public void runMode1(java.util.Collection<Rule> rules)

It takes a sample of unique predictions for the given rules by taking care of duplicated predictions. If a prediction was generated before by a rule, then it is not considered in the sampling.

Parameters:

rules -

runMode2

public void runMode2(java.util.Collection<Rule> rules)

It takes a sample of the predictions of every rule in the argument. Unlike the method runMode1, the result can contain duplicates if two rules predict the same fact and this fact is considered in both samples.

Parameters:

rules -

sample

public static java.util.Collection<Triple<ByteString,ByteString,ByteString>> sample(java.util.Collection<Triple<ByteString,ByteString,ByteString>> samplingCandidates,
                                                                                    int sampleSize)

Given a collection of predictions, it extracts a random sample from it.

Parameters:

samplingCandidates -

Returns:

main

public static void main(java.lang.String[] args)
                 throws java.io.IOException

Parameters:

args -

Throws:

java.io.IOException