Ejemplo n.º 1
0
    def create_loss_distribution(self, name, signal_id=1, file_name=''):
        test_bgs_data, test_signal_data = self.load_test_data(signal_id=signal_id)

        predict_bgs_test = self.predict(test_bgs_data)
        predict_signal_test = self.predict(test_signal_data)

        test_bgs_distribution = model_utils.loss_distribution(test_bgs_data,
                                                              predict_bgs_test)

        test_signal_distribution = model_utils.loss_distribution(test_signal_data,
                                                                 predict_signal_test)

        model_utils.plot_histogram(test_bgs_distribution.numpy(), test_signal_distribution.numpy(), name, file_name=file_name)
Ejemplo n.º 2
0
    def create_loss_distribution(self, name, m_5=6000, k=1000, file_name=''):
        test_bgs_data, test_signal_data = self.load_test_data(m_5=m_5, k=k)

        predict_bgs_test = self.predict(test_bgs_data)
        predict_signal_test = self.predict(test_signal_data)

        test_bgs_distribution = model_utils.loss_distribution(test_bgs_data,
                                                              predict_bgs_test.reshape(test_bgs_data.shape))

        test_signal_distribution = model_utils.loss_distribution(test_signal_data,
                                                                 predict_signal_test.reshape(test_signal_data.shape))
        model_utils.plot_histogram(test_bgs_distribution.numpy(),
                                   test_signal_distribution.numpy(),
                                   name,
                                   file_name=file_name)