TEST(NVDMPredictorTest, all) {
  std::vector<float> f;
  std::vector<std::vector<float>> input;

  std::string text(
      "facebook be bold and move fast connect people around the "
      "world facebook");

  std::vector<int> feature = featurizer->genFeature(text);
  for (const auto num : feature) {
    f.push_back(folly::to<float>(num));
  }

  input.push_back(f);
  std::vector<std::vector<float>> res = nvdm->predict(input);
  EXPECT_EQ(10000, feature.size());
  EXPECT_EQ(1, res.size());
  EXPECT_EQ(400, res[0].size());
}