use conda env yaml's for training-tuning-scripts

models/training-tuning-scripts/abp-models/requirements.yml

@@ -0,0 +1,25 @@
+# SPDX-FileCopyrightText: Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+channels:
+    - rapidsai
+    - nvidia
+    - conda-forge
+dependencies:
+    - cuml=23.06
+    - jupyterlab
+    - python=3.10
+    - scikit-learn=1.2.2
+    - xgboost

models/training-tuning-scripts/dfp-models/requirements.yml

@@ -0,0 +1,29 @@
+# SPDX-FileCopyrightText: Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+channels:
+    - nvidia
+    - pytorch
+    - conda-forge
+dependencies:
+    - dill
+    - jupyterlab
+    - matplotlib
+    - pandas=1.3
+    - python=3.10
+    - pytorch-cuda=11.8
+    - pytorch=2.0.1
+    - scikit-learn=1.2.2
+    - seaborn

models/training-tuning-scripts/fraud-detection-models/requirements.yml

@@ -16,15 +16,18 @@
 channels:
     - rapidsai
     - nvidia
+    - dglteam/label/cu118
+    - pytorch
     - conda-forge
 dependencies:
-    - click==8.1.3
+    - click>=8
     - cuml=23.06
-    - dgl==1.0.2+cu118
-    - numpy==1.23.5
-    - pandas==1.5.3
-    - scikit_learn==1.2.2
-    - torch==2.0.0+cu118
-    - torchmetrics==0.11.4
-    - tqdm==4.65.0
-    - xgboost==1.7.1
+    - dgl
+    - jupyterlab
+    - matplotlib
+    - pytorch-cuda=11.8
+    - pytorch=2.0.1
+    - scikit-learn=1.2.2
+    - tqdm=4
+    - torchmetrics
+    - xgboost

models/training-tuning-scripts/log-parsing-models/requirements.yml

@@ -0,0 +1,30 @@
+# SPDX-FileCopyrightText: Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+channels:
+    - rapidsai
+    - nvidia
+    - pytorch
+    - conda-forge
+dependencies:
+    - cudf=23.06
+    - jupyterlab
+    - onnx
+    - python=3.10
+    - pytorch-cuda=11.8
+    - pytorch=2.0.1
+    - seqeval=1.2.2
+    - tqdm=4
+    - transformers

models/training-tuning-scripts/phishing-models/requirements.yml

@@ -0,0 +1,30 @@
+# SPDX-FileCopyrightText: Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+channels:
+    - rapidsai
+    - nvidia
+    - pytorch
+    - conda-forge
+dependencies:
+    - cudf=23.06
+    - jupyterlab
+    - onnx
+    - python=3.10
+    - pytorch-cuda=11.8
+    - pytorch=2.0.1
+    - scikit-learn=1.2.2
+    - tqdm=4
+    - transformers

models/training-tuning-scripts/ransomware-models/requirements.yml

@@ -0,0 +1,24 @@
+# SPDX-FileCopyrightText: Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+channels:
+    - conda-forge
+dependencies:
+    - jupyterlab
+    - matplotlib
+    - pandas=1.3
+    - python=3.10
+    - scikit-learn=1.2.2
+    - seaborn

models/training-tuning-scripts/root-cause-models/requirements.yml

@@ -0,0 +1,30 @@
+# SPDX-FileCopyrightText: Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+channels:
+    - rapidsai
+    - nvidia
+    - pytorch
+    - conda-forge
+dependencies:
+    - cudf=23.06
+    - jupyterlab
+    - onnx
+    - python=3.10
+    - pytorch-cuda=11.8
+    - pytorch=2.0.1
+    - scikit-learn=1.2.2
+    - tqdm=4
+    - transformers

models/training-tuning-scripts/sid-models/requirements.yml

@@ -0,0 +1,30 @@
+# SPDX-FileCopyrightText: Copyright (c) 2022-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+channels:
+    - rapidsai
+    - nvidia
+    - pytorch
+    - conda-forge
+dependencies:
+    - cudf=23.06
+    - jupyterlab
+    - onnx
+    - python=3.10
+    - pytorch-cuda=11.8
+    - pytorch=2.0.1
+    - scikit-learn=1.2.2
+    - tqdm=4
+    - transformers