コード例 #1
0
ファイル: main.py プロジェクト: jack-and-rozz/werewolf
    mode = params[1]
    if mode == "-parse" :
        log_parser = LogParser()
        if (argc <= 1):
            log_srcfilename = FilePath.ROOTPATH + FilePath.SOURCE_LOG
            log_parser.getLog(log_srcfilename)
            parsed_logfilename = FilePath.ROOTPATH + FilePath.PARSED_LOG
        elif (argc == 2):
            print "error : 出力ファイル名を指定してください\n"
            exit()
        elif (argc >= 3):
            srcfile = FilePath.ROOTPATH + FilePath.LOGFILES + params[2]
            log_parser.getLog(srcfile)
            parsed_logfilename = FilePath.ROOTPATH + FilePath.LOGFILES + params[3]

        log_parser.saveParsedLog(parsed_logfilename,"w+")
    elif mode == "-learn":
        
        log_parser = LogParser()
        log_parser.loadParsedLog(FilePath.ROOTPATH + FilePath.PARSED_LOG)
        learning_engine = RoleInferenceEngine(log_parser)
        learning_engine.learnFromLog()
        learning_engine.checkEvaluations()
        learning_engine.saveParameters()

    elif mode == "-infer":
        role = params[2]
        role_inference_engine = RoleInferenceEngine()
        role_inference_engine.loadLearnedParameters()
        role_inference_engine.roleInferenceTest(role)
        #log_parser = LogParser()