ldsCtrlEst_h/lds_gaussian_fit_em.h #

//===-- ldsCtrlEst_h/lds_gaussian_fit_em.h - GLDS Fit (EM) ------*- C++ -*-===//
//
// Copyright 2021 Michael Bolus
// Copyright 2021 Georgia Institute of Technology
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
//===----------------------------------------------------------------------===//
//===----------------------------------------------------------------------===//

#ifndef LDSCTRLEST_LDS_GAUSSIAN_FIT_EM_H
#define LDSCTRLEST_LDS_GAUSSIAN_FIT_EM_H

#include "lds_fit_em.h"
#include "lds_gaussian_fit.h"

namespace lds {
namespace gaussian {
class FitEM : public EM<Fit> {
 public:
  using EM<Fit>::EM;

 private:
  void MaximizeOutput() override;

  void MaximizeMeasurement() override;

  void RecurseKe(Matrix& Ke, Cube& P_pre, Cube& P_post, size_t t) override;
};

}  // namespace gaussian
}  // namespace lds

#endif