def test_train_skipgram_model_default(self):
default_args = default_params.read_file(params_txt)
model = ft.skipgram(input_file, output)
# Make sure the default params of skipgram is equal
# to fasttext(1) default params
self.assertEqual(model.model_name, 'skipgram')
self.assertEqual(model.dim, int(default_args['dim']))
self.assertEqual(model.ws, int(default_args['ws']))
self.assertEqual(model.epoch, int(default_args['epoch']))
self.assertEqual(model.min_count, int(default_args['minCount']))
self.assertEqual(model.neg, int(default_args['neg']))
self.assertEqual(model.word_ngrams, int(default_args['wordNgrams']))
self.assertEqual(model.loss_name, default_args['loss'])
self.assertEqual(model.bucket, int(default_args['bucket']))
self.assertEqual(model.minn, int(default_args['minn']))
self.assertEqual(model.maxn, int(default_args['maxn']))
self.assertEqual(model.lr_update_rate,
float(default_args['lrUpdateRate']))
self.assertEqual(model.t, float(default_args['t']))
def test_train_cbow_model_default(self):
default_args = default_params.read_file(params_txt)
model = ft.cbow(input_file, output)
# Make sure the default params of cbow is equal
# to fasttext(1) default params
self.assertEqual(model.model_name, 'cbow')
self.assertEqual(model.dim, int(default_args['dim']))
self.assertEqual(model.ws, int(default_args['ws']))
self.assertEqual(model.epoch, int(default_args['epoch']))
self.assertEqual(model.min_count, int(default_args['minCount']))
self.assertEqual(model.neg, int(default_args['neg']))
self.assertEqual(model.word_ngrams, int(default_args['wordNgrams']))
self.assertEqual(model.loss_name, default_args['loss'])
self.assertEqual(model.bucket, int(default_args['bucket']))
self.assertEqual(model.minn, int(default_args['minn']))
self.assertEqual(model.maxn, int(default_args['maxn']))
self.assertEqual(model.lr_update_rate,
float(default_args['lrUpdateRate']))
self.assertEqual(model.t, float(default_args['t']))
