Medial Code Documentation
Loading...
Searching...
No Matches
MedSVM.h
1#pragma once
2
4
5class MedSvm : public MedPredictor {
6public:
7 // Model
8
9 struct svm_parameter params;
10 struct svm_model *model;
11 /*double **x;
12 double **y;
13 float *w;
14 */
15
16 // Function
17 MedSvm();
18 MedSvm(void *params);
19 MedSvm(struct svm_parameter &params);
20 ~MedSvm();
21
22 void init_defaults();
23 int init(void *params);
26 virtual int set_params(map<string, string>& mapper);
27 int init(struct svm_parameter &params);
28
29 int Learn(float *x, float *y, const float *w, int nsamples, int nftrs);
30 int Predict(float *x, float *&preds, int nsamples, int nftrs) const;
31
32 ADD_CLASS_NAME(MedSvm)
33 size_t get_size();
34 size_t serialize(unsigned char *blob);
35 size_t deserialize(unsigned char *blob);
36
37private:
38
39
40};
MedAlgo - APIs to different algorithms: Linear Models, RF, GBM, KNN, and more.
Base Interface for predictor.
Definition MedAlgo.h:78
Definition MedSVM.h:5
int Learn(float *x, float *y, const float *w, int nsamples, int nftrs)
Learn should be implemented for each model.
Definition MedSvm.cpp:87
int Predict(float *x, float *&preds, int nsamples, int nftrs) const
Predict should be implemented for each model.
Definition MedSvm.cpp:120
virtual int set_params(map< string, string > &mapper)
The parsed fields from init command.
Definition MedSvm.cpp:30
size_t deserialize(unsigned char *blob)
Deserialiazing blob to object. returns number of bytes read.
Definition MedSvm.cpp:146
size_t serialize(unsigned char *blob)
Serialiazing object to blob memory. return number ob bytes wrote to memory.
Definition MedSvm.cpp:143
size_t get_size()
Gets bytes sizes for serializations.
Definition MedSvm.cpp:140
Definition svm.h:53
Definition svm.h:29