Skip to content

Commit 6e71bc5

Browse files
authored
Compatibility with JENN 2.0 (#1282)
* regenerate pkl model files with jenn 2.0 * unpin jenn
1 parent bb11943 commit 6e71bc5

5 files changed

Lines changed: 2 additions & 5 deletions

File tree

Binary file not shown.
Binary file not shown.

examples/other_files/ML_AI_Plugin/Other_MLAI_Models/mea_column_model_training_jenn.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,7 @@
1515
import numpy as np
1616
import pandas as pd
1717

18-
# from smt.utils.neural_net.model import Model
19-
from jenn.model import NeuralNet
18+
from jenn import NeuralNet
2019
import pickle
2120
from types import SimpleNamespace
2221

examples/other_files/ML_AI_Plugin/Other_MLAI_Models/mea_column_model_training_smtgenn.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,8 +15,6 @@
1515
import numpy as np
1616
import pandas as pd
1717

18-
# from smt.utils.neural_net.model import Model
19-
from jenn.model import NeuralNet
2018
from smt.surrogate_models import GENN
2119
import pickle
2220
from types import SimpleNamespace

requirements-mlai.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@ sympy
33
torch
44
scikit-learn
55
smt
6-
jenn<2.0.0 # Temp fix until we support JENN 2, see issue #1276
6+
jenn

0 commit comments

Comments
 (0)