block_update and install in local folder

This commit is contained in:
2024-04-10 00:55:36 +02:00
parent 1326891d6a
commit cf9b5716ac
7 changed files with 90 additions and 18 deletions

View File

@@ -25,6 +25,7 @@ namespace bayesnet {
void buildModel(const torch::Tensor& weights) override;
void trainModel(const torch::Tensor& weights) override;
private:
std::tuple<torch::Tensor&, double, bool> update_weights_block(int k, torch::Tensor& ytrain, torch::Tensor& weights);
std::vector<int> initializeModels();
torch::Tensor X_train, y_train, X_test, y_test;
// Hyperparameters
@@ -36,7 +37,7 @@ namespace bayesnet {
std::string select_features_algorithm = Orders.DESC; // Selected feature selection algorithm
FeatureSelect* featureSelector = nullptr;
double threshold = -1;
bool block_update = true;
bool block_update = false;
};
}
#endif