private static void TestPatternBased(string question_filename, string output_filename) { StreamReader sr = new StreamReader(question_filename); PatternBased pb = new PatternBased(); StreamWriter sw = new StreamWriter(output_filename); Parser parse = new Parser(); while (!sr.EndOfStream) { PatternResponse pr = new PatternResponse(); string line = sr.ReadLine(); string question = line.Split(new char[] { '\t' }, StringSplitOptions.RemoveEmptyEntries)[0]; Query query = new Query(question); parse.PosTagging(ref query); if (pb.QuestionClassify(query, out pr)) { sw.Write(question); sw.Write('\t'); sw.WriteLine(pr.property); } else { sw.WriteLine(question); } } sw.Flush(); sw.Close(); }
//GraphEngineQuery graphengine_query = new GraphEngineQuery(); public static bool DoKBQA(Query query, Parser parser) { parser.ParseAllTag(ref query); PatternResponse pattern_response; if (pattern_qa.QuestionClassify(query, out pattern_response))// || cnn_qa.QuestionClassify(query, out pattern_response)) { Console.WriteLine("Start to KBQA"); string question_topic = ""; try { switch (pattern_response.entity_type) { case KBQAEntityType.Movie: question_topic = query.carried_movie[0]; break; case KBQAEntityType.Celebrity: question_topic = (query.carried_artist.Count > 0) ? query.carried_artist[0] : query.carried_director[0]; break; case KBQAEntityType.RecentMovie: question_topic = ""; break; case KBQAEntityType.IsPublish: question_topic = query.carried_movie[0]; break; } //List<object> res = graphengine_query.GetGraphEngineData(question_topic, pattern_response.property, pattern_response.hop_num); List <string> res = GetColumnData(pattern_response.entity_type, question_topic, pattern_response.property); res = res.Distinct().ToList(); string answer = string.Join(",", res.ToArray()); if (answer.Length < 2) { Utils.WriteUnknow("数据库中没有相关的答案...", query.raw_query); } else { Utils.WriteMachine(answer); } return(true); } catch (Exception e) { Utils.WriteUnknow("It seems Neural Network makes a mistake", query.raw_query); return(false); } } else { return(false); } }
private static void TestGraphEngineQA() { //string question = "肯尼思·洛纳根导演过哪些电影"; // 肯尼思·洛纳根 splited //string question = "你的名字是哪个国家拍的"; // 你的名字 in NER, but 你的名字。in CellID //string question = "十二怒汉是讲什么的"; // have no 十二怒汉 //string question = "活着是讲什么的"; //string question = "你的名字。是讲什么的"; // the period //string question = "赌神是讲什么的"; //string question = "天下无贼是谁导演的"; string question = "林家栋拍过什么电影"; //拍 act? direct? //string question = "大话西游之大圣娶亲是什么时候拍的"; //string question = "有木有徐克的"; Parser parser = new Parser(); GraphEngineQuery graphengine_query = new GraphEngineQuery(); PatternBased pattern_qa = new PatternBased(); Query query = new Query(question); parser.PosTagging(ref query); parser.ParseAllTag(ref query); PatternResponse pattern_response; if (pattern_qa.QuestionClassify(query, out pattern_response)) { string question_topic = ""; switch (pattern_response.entity_type) { case KBQAEntityType.Movie: question_topic = query.carried_movie[0]; break; case KBQAEntityType.Celebrity: question_topic = (query.carried_artist.Count > 0) ? query.carried_artist[0] : query.carried_director[0]; break; } List <object> res = graphengine_query.GetGraphEngineData(question_topic, pattern_response.property, pattern_response.hop_num); Utils.WriteMachine(string.Join(",", res.ToArray())); } }