public GloveLearningModel(IMerkleTreeContext context)
{
m_context = context;
cooccurArgs = new CooccurArgs();
FillCooccurArgs(ref cooccurArgs);
gloveArgs = new GloveArgs();
FillGloveArgs(ref gloveArgs);
shuffleArgs = new ShuffleArgs();
FillShuffleArgs(ref shuffleArgs);
vocabCountArgs = new VocabCountArgs();
FillVocabCountArgs(ref vocabCountArgs);
}
public void FillGloveArgs(ref GloveArgs args)
{
args.verbose = 1;
args.vectorSize = 50;
args.threads = 8;
args.iter = 15;
args.eta = 0.05f;
args.alpha = 0.75f;
args.xMax = 100.0f;
args.binary = 0;
args.model = 2;
args.saveGradsq = 0;
args.checkpointEvery = 0;
args.mode = 0;
}
static extern int glove(ref GloveArgs args, string shufCooccurInFile, string vocabInFile, string gloveOutFile, string gradsqOutFile);