aimodelshare 0.1.29__py3-none-any.whl → 0.1.64__py3-none-any.whl

aimodelshare-0.1.29.dist-info/licenses/LICENSE

@@ -1,22 +0,0 @@
-
-MIT License
-
-Copyright (c) 2023 AI Model Share Initiative (And all affiliated organizations and individuals)
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.

tests/test_aimsonnx.py

@@ -1,135 +0,0 @@
-from aimodelshare.aimsonnx import _get_layer_names
-from aimodelshare.aimsonnx import _get_layer_names_pytorch
-from aimodelshare.aimsonnx import _get_sklearn_modules
-from aimodelshare.aimsonnx import model_from_string
-from aimodelshare.aimsonnx import _get_pyspark_modules
-from aimodelshare.aimsonnx import pyspark_model_from_string
-from aimodelshare.aimsonnx import layer_mapping
-from aimodelshare.aimsonnx import _sklearn_to_onnx
-from aimodelshare.aimsonnx import _pyspark_to_onnx
-from aimodelshare.aimsonnx import _keras_to_onnx
-from aimodelshare.aimsonnx import _pytorch_to_onnx
-from aimodelshare.aimsonnx import _misc_to_onnx
-from sklearn.linear_model import LogisticRegression
-from sklearn.neural_network import MLPClassifier
-import onnx
-from xgboost import XGBClassifier
-from pyspark.ml.classification import RandomForestClassifier, MultilayerPerceptronClassifier
-from keras.models import Sequential
-from torch import nn
-import torch
-from tensorflow.keras.layers import Dense
-
-def test_sklearn_to_onnx():
-
-    from sklearn.datasets import load_iris
-    data = load_iris()
-    X = data.data
-    y = data.target
-
-    model = LogisticRegression(C=10, penalty='l1', solver='liblinear')
-    model.fit(X, y)
-    onnx_model = _sklearn_to_onnx(model)
-    assert isinstance(onnx_model, onnx.ModelProto)
-
-    # model = MLPClassifier()
-    # model.fit(X, y)
-    # onnx_model = _sklearn_to_onnx(model)
-    # assert isinstance(onnx_model, onnx.ModelProto)
-
-
-# def test_misc_to_onnx():
-#
-#     model = XGBClassifier()
-#     onnx_model = _misc_to_onnx(model)
-#     assert isinstance(onnx_model, onnx.ModelProto)
-
-
-# def test_pyspark_to_onnx():
-#
-#     model =RandomForestClassifier(labelCol="indexedLabel", featuresCol="indexedFeatures", numTrees=10)
-#     onnx_model = _pyspark_to_onnx(model)
-#     assert isinstance(onnx_model, onnx.ModelProto)
-#
-#     model = MultilayerPerceptronClassifier()
-#     onnx_model = _pyspark_to_onnx(model)
-#     assert isinstance(onnx_model, onnx.ModelProto)
-
-def test_keras_to_onnx():
-
-    model = Sequential()
-    model.add(Dense(12, input_shape=(8,), activation='relu'))
-    model.add(Dense(8, activation='relu'))
-    model.add(Dense(1, activation='sigmoid'))
-    model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])
-
-    onnx_model = _keras_to_onnx(model)
-    assert isinstance(onnx_model, onnx.ModelProto)
-
-
-def test_pytorch_to_onnx():
-
-    model = nn.Sequential(nn.Linear(3, 3),
-                          nn.ReLU(),
-                          nn.Linear(3, 1),
-                          nn.Sigmoid())
-
-    onnx_model = _pytorch_to_onnx(model, torch.randn(1, 3))
-    assert isinstance(onnx_model, onnx.ModelProto)
-
-
-def test_get_layer_names():
-
-    layers = _get_layer_names()
-
-    assert isinstance(layers, tuple)
-
-
-def test_get_layer_names_pytorch():
-
-        layers = _get_layer_names_pytorch()
-
-        assert isinstance(layers, tuple)
-
-
-def test_get_sklearn_modules():
-
-        modules = _get_sklearn_modules()
-
-        assert isinstance(modules, dict)
-
-def test_model_from_string():
-
-    model_class = model_from_string("RandomForestClassifier")
-
-    assert model_class.__name__ == "RandomForestClassifier"
-
-
-def test_get_pyspark_modules():
-
-    modules = _get_pyspark_modules()
-
-    assert isinstance(modules, dict)
-
-
-def test_pyspark_model_from_string():
-
-    model_class = pyspark_model_from_string("RandomForestClassifier")
-
-    assert model_class.__name__ == "RandomForestClassifier"
-
-
-def test_layer_mapping():
-
-    layer_map = layer_mapping(direction="torch_to_keras")
-    assert isinstance(layer_map, dict)
-
-    layer_map = layer_mapping(direction="keras_to_torch")
-    assert isinstance(layer_map, dict)
-
-    layer_map = layer_mapping(direction="torch_to_keras", activation=True)
-    assert isinstance(layer_map, dict)
-
-    layer_map = layer_mapping(direction="keras_to_torch", activation=True)
-    assert isinstance(layer_map, dict)
-