public void CountPrepareNetwork()
{
try
{
CountNetworkTrainController.PrepareNetwork(ref container2, ref trainSettings);
}
catch
{
Assert.Fail();
}
Assert.Pass();
}
public void CountIncorrectInputDirectory()
{
CountNetworkTrainSettings trainSettings2 = new CountNetworkTrainSettings
{
correctInputDirectory = @"../NoLocation",
incorrectInputDirectory = @"../../../IncorrectData",
outputDirectory = dataOutLocation,
outputName = dataOutName,
sampleCount = 10
};
Assert.Throws <System.IO.DirectoryNotFoundException>(() => { CountNetworkTrainController.PrepareData(ref container2, ref trainSettings2); });
}
static void Main(string[] args)
{
// Main program.
// A Network container is used to control and hold a network
NetworkContainer container = new NetworkContainer
{
verbose = true
};
// Used to declare settings for a particular network
CountNetworkTrainSettings trainSettings = new CountNetworkTrainSettings
{
correctInputDirectory = @"./Data_Correct",
incorrectInputDirectory = @"./Data_Incorrect",
outputDirectory = @"./Data_Out",
outputName = "DemoNetwork",
sampleCount = 20
};
CountNetworkTrainController.PrepareData(ref container, ref trainSettings);
CountNetworkTrainController.PrepareNetwork(ref container, ref trainSettings);
CountNetworkTrainController.Train(ref container, ref trainSettings);
CountNetworkPredictSettings predictSettings = new CountNetworkPredictSettings
{
trainedNetwork = @"./Data_Out/DemoNetwork.eg",
predictData = @"./Data_Predict/good.csv",
sampleCount = 20,
predictSettings = new EncogPredictSettings
{
threshold = 0.92
}
};
CountNetworkPredictController.PreparePredictor(ref container, ref predictSettings);
Console.WriteLine("The given file is: " + CountNetworkPredictController.Predict(ref container, ref predictSettings));
}
public void CountPrepareNetworkBeforePrepareData()
{
Assert.Throws <IncorrectActionOrderException>(() => { CountNetworkTrainController.PrepareNetwork(ref container2, ref trainSettings); });
}
public void CountTrainAndSaveNetwork()
{
CountNetworkTrainController.Train(ref container2, ref trainSettings);
Assert.IsTrue(File.Exists(dataOutLocation + dataOutName + ".eg"));
}
