Compare commits
2 Commits
56236d755f
...
065de4c5e8
Author | SHA1 | Date |
---|---|---|
|
065de4c5e8 | |
|
db226aee60 |
|
@ -361,7 +361,8 @@ class eNoseTrainer:
|
||||||
plt.close()
|
plt.close()
|
||||||
|
|
||||||
def fit(self):
|
def fit(self):
|
||||||
windows = [128, 256, 384]
|
windows = [128,]
|
||||||
|
# windows = [128, 256, 384]
|
||||||
total_train_queue = 2*int(1/self.ratio)*(len(self.get_model_train())+len(windows))
|
total_train_queue = 2*int(1/self.ratio)*(len(self.get_model_train())+len(windows))
|
||||||
self.logger.info("{:=^60}".format(f'Begin Fit {total_train_queue} Models'))
|
self.logger.info("{:=^60}".format(f'Begin Fit {total_train_queue} Models'))
|
||||||
self.trained = 0
|
self.trained = 0
|
||||||
|
|
|
@ -13,7 +13,7 @@ fi
|
||||||
|
|
||||||
env_name=$1
|
env_name=$1
|
||||||
|
|
||||||
conda create -n "$env_name" python scikit-learn==1.3.1 xgboost conda-forge::tensorflow-cpu conda-forge::ray-tune keras pandas numpy matplotlib openpyxl xlsxwriter conda-forge::enlighten
|
conda create -n "$env_name" python scikit-learn==1.3.1 xgboost conda-forge::tensorflow-cpu conda-forge::optuna keras pandas numpy matplotlib openpyxl xlsxwriter conda-forge::enlighten
|
||||||
if [ $? -eq 0 ]; then
|
if [ $? -eq 0 ]; then
|
||||||
echo "Packages installed successfully in environment '$env_name'"
|
echo "Packages installed successfully in environment '$env_name'"
|
||||||
else
|
else
|
||||||
|
|
Loading…
Reference in New Issue