def main(): os.chdir(os.path.dirname(sys.argv[0])) if not os.path.exists('./pickles/features-data.pickle'): featuresDataFile = open('./pickles/features-data.pickle', 'wb') featuresData = FeaturesData() featuresData.loadMask(maskFn='./resources/features/mask.csv') pickle.dump(featuresData, featuresDataFile) featuresDataFile.close() else: featuresDataFile = open('./pickles/features-data.pickle', 'rb') featuresData = pickle.load(featuresDataFile) featuresDataFile.close() if not os.path.exists('./pickles/users-data.pickle'): usersDataFile = open('./pickles/users-data.pickle', 'wb') usersData = UsersData() pickle.dump(usersData, usersDataFile) usersDataFile.close() else: usersDataFile = open('./pickles/users-data.pickle', 'rb') usersData = pickle.load(usersDataFile) usersDataFile.close() combinedData = CombinedData(featuresData, usersData) voteGuesser = VoteGuesser(combinedData, k=3) voteGuesser.solveTask(outputFn='./output/task1.csv')
def main(): os.chdir(os.path.dirname(sys.argv[0])) """Zadanie 3.""" """ featuresData = FeaturesData(featuresFn='./resources/features-zadanie3/feature.csv', dataFn='./resources/features-zadanie3/data.csv', weightsFn='./resources/features-zadanie3/weight-manual.csv') voteGuesser = VoteGuesser(featuresData, k=3) print('Average error:', voteGuesser.calculateReguessExpectedError()) voteGuesser.solveTask(outputFn='./output/zadanie3/task.csv') """ """Zadanie 4.""" featuresData = FeaturesData(featuresFn='./resources/features-zadanie4/feature.csv', dataFn='./resources/features-zadanie4/data.csv', weightsFn='./resources/features-zadanie4/weight.csv') voteGuesser = VoteGuesser(featuresData, k=3) print('Average error for all features:', voteGuesser.calculateReguessExpectedError()) voteGuesser.solveTask(outputFn='./output/zadanie4/task1.csv') voteGuesser.loadMask(maskFn='./output/zadanie4/mask.csv') print('Average error for selected features:', voteGuesser.calculateReguessExpectedError()) voteGuesser.solveTask(outputFn='./output/zadanie4/task2.csv')