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);