CosanModel.h

Go to the documentation of this file.

//
// Created by Xinyu Zhang on 3/27/21.
//
 
#ifndef COSAN_COSANMODEL_H
#define COSAN_COSANMODEL_H
#include <chrono>
#include <cosan/base/CosanBO.h>
namespace Cosan{
    /**
     * Problem Type.
     **/
    enum EProblemType{
        PdNone   = 0,
        PdBinaryClassification = 1,
        PdUnivariateRegression=2,
    };
    /**
     * Model Type
     */
    enum EModelType{
        MdNone = 0,
        MdLinearRegression=1,
        MdRidgeRegression=2,
        MdPrincipalComponentRegression=3,
        MdRidgePrincipalComponentRegression=4
    };
    /**
     * Solver Type
     */
    enum ESolverType{
        SrAuto = 0,
        ADMM = 1
 
    };
    /**
     * @brief Base Object for CosanModel
     */
    class CosanModel: public CosanBO {
    public:
        CosanModel(): CosanBO(),MaxTrainTime(std::chrono::seconds(std::chrono::hours(1)).count()),
        SolverType(SrAuto){
        }
//        virtual ~CosanModel();
//        virtual Response* predict(const Feature* X = NULL)=0;//fixed later
 
        virtual void SetMaxTrainTime(double t) { MaxTrainTime=t;}
        virtual double GetMaxTrainTime() const {return MaxTrainTime;}
 
        virtual EModelType GetModelType() {return MdNone;}
        virtual EProblemType GetProblemType(){return PdNone;}
        virtual ESolverType GetSolverType() const {return SolverType;}
        void SetSolverType(ESolverType sr) {SolverType=sr;}
    protected:
        double MaxTrainTime;
        ESolverType SolverType;
    };
}
 
#endif //COSAN_COSANMODEL_H