Sync public subset from Flux
This commit is contained in:
@@ -11,7 +11,7 @@
|
||||
namespace neural_networks{
|
||||
|
||||
template <typename TX, typename Ty>
|
||||
void create_spital_data(const uint64_t samples, const uint64_t classes, utils::Matrix<TX>& X, utils::Vector<Ty>& y) {
|
||||
void create_spital_data(const uint64_t samples, const uint64_t classes, utils::Matrix<TX>& X, utils::Matrix<Ty>& y) {
|
||||
|
||||
const uint64_t rows = samples*classes;
|
||||
TX r, t;
|
||||
@@ -21,8 +21,8 @@ namespace neural_networks{
|
||||
if ((rows != X.rows()) || (X.cols() != 2)){
|
||||
X.resize(samples*classes, 2);
|
||||
}
|
||||
if (rows != y.size()){
|
||||
y.resize(rows);
|
||||
if (rows != y.rows()){
|
||||
y.resize(rows, 1);
|
||||
}
|
||||
|
||||
for (uint64_t i = 0; i < classes; ++i){
|
||||
@@ -33,7 +33,7 @@ namespace neural_networks{
|
||||
|
||||
X(row_idx, 0) = r*std::cos(t*2.5) + utils::random(TX{-0.15}, TX{0.15});
|
||||
X(row_idx, 1) = r*std::sin(t*2.5) + utils::random(TX{-0.15}, TX{0.15});
|
||||
y[row_idx] = static_cast<Ty>(i);
|
||||
y(row_idx, 0) = static_cast<Ty>(i);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user