update notebook + version update (opensearch-project#76)

* update notebook + version update

Signed-off-by: Dhrubo Saha <[email protected]>

* updated docs

Signed-off-by: Dhrubo Saha <[email protected]>

---------

Signed-off-by: Dhrubo Saha <[email protected]>

Author: dhrubo-os

docs/source/examples/demo_tracing_model_torchscript_onnx.ipynb

@@ -546,4 +546,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 5
-}
+}

docs/source/examples/demo_transformer_model_train_save_upload_to_openSearch.ipynb

@@ -58,7 +58,31 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 3,
+   "id": "e239ffc9",
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "('generate.py', <http.client.HTTPMessage at 0x10bdbe940>)"
+      ]
+     },
+     "execution_count": 3,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "# Download generate.py for Generate Synthetic Queries\n",
+    "\n",
+    "import urllib.request\n",
+    "urllib.request.urlretrieve(\"https://artifacts.opensearch.org/models/ml-models/huggingface/sentence-transformers/msmarco-distilbert-base-tas-b/1.0.0/onnx/config.json\", \"generate.py\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 11,
    "id": "17a3e085",
    "metadata": {
     "scrolled": true
@@ -70,7 +94,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": 4,
    "id": "87c021df",
    "metadata": {},
    "outputs": [],
@@ -93,7 +117,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 5,
    "id": "798cac25",
    "metadata": {},
    "outputs": [],
@@ -104,7 +128,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 6,
    "id": "5c85ae17",
    "metadata": {},
    "outputs": [],
@@ -114,7 +138,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 7,
    "id": "77442abf",
    "metadata": {},
    "outputs": [],
@@ -137,7 +161,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 8,
    "id": "89e1cb2a",
    "metadata": {},
    "outputs": [],
@@ -147,7 +171,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 8,
+   "execution_count": 9,
    "id": "9bc33a2d",
    "metadata": {},
    "outputs": [],
@@ -519,4 +543,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 5
-}
+}

docs/source/examples/online_retail_analysis.ipynb

@@ -7,7 +7,11 @@
      "name": "#%% md\n"
     }
    },
-   "source": []
+   "source": [
+    "# Demo Notebook for Online Retail Analysis\n",
+    "\n",
+    "#### [download notebook](https://github.com/opensearch-project/opensearch-py-ml/blob/main/docs/source/examples/online_retail_analysis.ipynb)"
+   ]
   },
   {
    "cell_type": "markdown",
@@ -1702,4 +1706,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
+}

docs/source/reference/api/sentence_transformer.load_sentence_transformer_example.rst

@@ -0,0 +1,6 @@
+SentenceTransformerModel.load_training_data
+==============================================
+
+.. currentmodule:: opensearch_py_ml
+
+.. autofunction:: opensearch_py_ml.ml_models.SentenceTransformerModel.load_training_data

docs/source/reference/api/sentence_transformer.load_training_data.rst

@@ -1,5 +1,5 @@
 SentenceTransformerModel.save_as_onnx
-===================================
+=======================================
 
 .. currentmodule:: opensearch_py_ml
 

docs/source/reference/api/sentence_transformer.save_as_onnx.rst

@@ -25,7 +25,7 @@ Train Model
    :maxdepth: 2
 
    api/sentence_transformer.read_queries
-   api/sentence_transformer.load_sentence_transformer_example
+   api/sentence_transformer.load_training_data
    api/sentence_transformer.set_up_accelerate_config
    api/sentence_transformer.train_model
    api/sentence_transformer.zip_model

docs/source/reference/sentencetransformer.rst

@@ -27,7 +27,7 @@
 __title__ = "opensearch_py_ml"
 __description__ = "Python Client and Toolkit for DataFrames, Big Data, Machine Learning and ETL in OpenSearch"
 __url__ = "https://github.com/opensearch-project/opensearch-py-ml"
-__version__ = "1.0.0b1"
+__version__ = "1.0.0"
 __author__ = "Dhrubo Saha"
 __author_email__ = "[email protected], "
 __maintainer__ = "Dhrubo Saha, Yaliang Wu"

opensearch_py_ml/_version.py

@@ -469,7 +469,7 @@ def train_model(
                 num_training_steps=num_epochs * len(train_dataloader),
             )
 
-            accelerator = Accelerator(fp16=training_args.fp16)
+            accelerator = Accelerator()
             model, optimizer, train_dataloader, scheduler = accelerator.prepare(
                 model, optimizer, train_dataloader, scheduler
             )