private static ClassificationProblem CreateChessBoard()
{
ClassificationProblem problem = new ClassificationProblem();
CategoryCollection collect = new CategoryCollection();
collect.Add(new Category(+1, "+1"));
collect.Add(new Category(-1, "-1"));
problem.Dimension = 2;
problem.CategoryCollection = collect;
problem.TrainingSet = GetExamples(collect);
problem.ValidationSet = GetExamples(collect);
return problem;
}
/// <summary>
/// foamliu, 2009/12/21, please make sure you've uncompressed "2_newsgroups.7z" in the "data" folder.
/// </summary>
/// <returns></returns>
private static ClassificationProblem CreateText()
{
const string DataFolder = @"..\data\2_newsgroups";
ClassificationProblem problem = new ClassificationProblem();
ExampleSet t_set = new ExampleSet();
ExampleSet v_set = new ExampleSet();
CategoryCollection collect = new CategoryCollection();
collect.Add(new Category(+1, "+1"));
collect.Add(new Category(-1, "-1"));
problem.Dimension = 2;
problem.CategoryCollection = collect;
DirectoryInfo dataFolder = new DirectoryInfo(DataFolder);
DirectoryInfo[] subfolders = dataFolder.GetDirectories();
int count = 0;
for (int i = 0; i < subfolders.Count(); i++)
{
DirectoryInfo categoryFolder = subfolders[i];
int cat = i * 2 - 1;
// for all the text files in each category
FileInfo[] files = categoryFolder.GetFiles();
count = 0;
int trainSetCount = Convert.ToInt32(Constants.TrainingSetRatio * files.Count());
for (int j = 0; j < files.Count(); j++)
{
FileInfo textFile = files[j];
Example e = new Example();
if (++count < trainSetCount)
{
t_set.AddExample(e);
}
else
{
v_set.AddExample(e);
}
}
}
problem.TrainingSet = t_set;
problem.ValidationSet = v_set;
return problem;
}
