def train(self): if self.validate_files(): if self.validate_bins(): self._model = nb.NaiveBayseAlgorithm( int(self._entry_bins.get())) dataset = pd.read_csv('train.csv', sep=',') self._model.train(dataset) answer = messagebox.askokcancel( "Naive Bayes Classifier", "Building classifier using train-set is done!")