Datasets.hpp

//
// Created by mateus on 20/05/2021.
//
 
#pragma once
#ifndef M_PI
#define M_PI 3.14159265358979323846
#endif
 
#include "ufjfmltk/core/Data.hpp"
 
 
namespace mltk::datasets{
    using Centers = std::vector<mltk::Point<double>>;
 
    struct RegPair{
        mltk::Data<double> dataset;
        mltk::Point<double> coef;
    };
 
    struct BlobsPair{
        mltk::Data<double> dataset;
        Centers centers;
    };
 
    mltk::Data<double> make_spirals(size_t n_samples=100, int n_classes=2, bool shuffle=true, double noise=1.0,
                            size_t n_loops=2, double margin = 0.5, size_t seed = 0);
 
    BlobsPair make_blobs(size_t n_samples=100, int n_centers=2, int n_dims=2, double cluster_std=1.0,
                                  double center_min=-10.0, double center_max=10.0, bool shuffle=true,
                                  bool has_classes=true, size_t seed = 0);
 
    BlobsPair make_blobs(const std::vector<size_t>& n_samples, const std::vector<mltk::Point<double>>& centers,
                         std::vector<double> clusters_std, int n_dims=2, bool shuffle=true,
                         bool has_classes=true, size_t seed = 0);
 
    RegPair make_regression(size_t n_samples=100, size_t n_dims=100, double bias=0.0, double noise=0.1, double stdev=0.01,
                            size_t n_informative=10, bool shuffle=true, size_t seed=0);
 
}