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)