uk.ac.ed.inf.enzml.mulan.predict
Class MulanPredict
java.lang.Object
uk.ac.ed.inf.utils.Initialised
uk.ac.ed.inf.enzml.mulan.predict.MulanPredict
- All Implemented Interfaces:
- uk.ac.ed.inf.utils.Initialisable
- Direct Known Subclasses:
- MulanPredictWithTrainedModel
public class MulanPredict
- extends uk.ac.ed.inf.utils.Initialised
First trains on a dataset (training set) and then emits predictions for a
(compatible unlabelled) dataset.
- Version:
- 19 Sep 2010
- Author:
- Luna De Ferrari luna.deferrari-at-ed.ac.uk
| Fields inherited from class uk.ac.ed.inf.utils.Initialised |
m_initialisationIsCorrect |
|
Constructor Summary |
MulanPredict(int trainArffId,
int unlabeledArffId,
java.lang.String arffRecordsDatabase,
java.lang.String predictionsDirectory,
MulanLearner learner)
|
| Methods inherited from class uk.ac.ed.inf.utils.Initialised |
initialisationIsCorrect, setInitialisationIsCorrect |
| Methods inherited from class java.lang.Object |
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait |
INSTANCE_ID
public static final java.lang.String INSTANCE_ID
- See Also:
- Constant Field Values
INSTANCE_NAME
public static final java.lang.String INSTANCE_NAME
- See Also:
- Constant Field Values
ATTRIBUTE_ID
public static final java.lang.String ATTRIBUTE_ID
- See Also:
- Constant Field Values
ATTRIBUTE_NAME
public static final java.lang.String ATTRIBUTE_NAME
- See Also:
- Constant Field Values
CONFIDENCE
public static final java.lang.String CONFIDENCE
- See Also:
- Constant Field Values
MulanPredict
public MulanPredict(int trainArffId,
int unlabeledArffId,
java.lang.String arffRecordsDatabase,
java.lang.String predictionsDirectory,
MulanLearner learner)
generateInstancesNames
public java.util.Vector<java.lang.String> generateInstancesNames()
getInstances
public java.util.TreeSet<java.lang.String> getInstances(uk.ac.ed.inf.utils.database.DbReader dataReader,
java.lang.String instancesQuery)
getPredictionsFile
public java.lang.String getPredictionsFile()
- The full path to the predictions file
- Returns:
- full path to the predictions file
getTrainArff
public MulanArffRecord getTrainArff()
getUnlabeledArff
public MulanArffRecord getUnlabeledArff()
getUnlabeledData
public weka.core.Instances getUnlabeledData()
predict
public void predict()
processOutput
public java.lang.String processOutput(int instanceIndex,
mulan.classifier.MultiLabelOutput instanceOutput)
savePredictions
public void savePredictions()
setPredictionsAndLogFile
public void setPredictionsAndLogFile(java.lang.String predictionsDirectory)
main
public static void main(java.lang.String[] args)
predictionsHeader
public static java.lang.String predictionsHeader()
run
public static java.lang.String run(int trainId,
int testId,
java.lang.String machineLearningProps,
java.lang.String predictionsDirectory,
MulanLearner learner)
- Generates and saves predictions.
- Parameters:
trainId - testId - machineLearningProps - predictionsDirectory - learner -
- Returns:
- the path to the predictions file