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')
