#include <shogun/base/init.h>
#include <shogun/base/some.h>
#include <shogun/features/DenseFeatures.h>
#include <shogun/labels/MulticlassLabels.h>
#include <shogun/multiclass/GaussianNaiveBayes.h>
#include <shogun/io/SerializableAsciiFile.h>
#include <shogun/lib/SGVector.h>
#include <shogun/io/CSVFile.h>
#include <iostream>
#include <assert.h>

using namespace shogun;

int main(int, char*[])
{
init_shogun_with_defaults();

auto f_feats_train = some<CCSVFile>("/home/myamada/np_data/output_with_labels_remapped1000.csv.shogun");
//auto f_feats_test = some<CCSVFile>("../../data/classifier_4class_2d_linear_features_test.dat");
auto f_labels_train = some<CCSVFile>("/home/myamada/np_data/output_with_labels_remapped1000.csv.shogun.labels");

//![create_features]
auto features_train = some<CDenseFeatures<float64_t>>(f_feats_train);
//auto features_test = some<CDenseFeatures<float64_t>>(f_feats_test);
auto labels_train = some<CMulticlassLabels>(f_labels_train);
//![create_features]


//![create_instance]
auto gnb = some<CGaussianNaiveBayes>(features_train, labels_train);
//![create_instance]

//![train_and_apply]
gnb->train();
auto saveFile = some<CSerializableAsciiFile>("/home/myamada/tmp/shogun_model2.out", 'w');
gnb->save_serializable(saveFile);
exit_shogun();
return 0;
}