def predictSingle(self, songFile):
data_base = DataPreprocessing.DataPeprocessing(self.db_file_name, self.processed_data_file_name)
data_base.data_preprosessing(PreprocessingTypes.EQUAL_STEPS)
self.DAG = self.convertFileToDAG()
bn = BayesianNetwork.BN(self.DAG)
res = bn.BNForOneSong(self.DAG, self.processed_data_file_name, self.predicted_results_file_name, songFile)
print(res)
return res
def preprocessing(self):
data_base = DataPreprocessing.DataPeprocessing(self.db_file_name, self.processed_data_file_name)
"""
1. 'MeanShirf' - default
2. 'Uniform Distribution'
3. 'Equal Steps'
"""
data_base.data_preprosessing(PreprocessingTypes.UNIFORM_DISTRIBUTION)