ai-data-science-team 0.0.0.9014__tar.gz → 0.0.0.9016__tar.gz

{ai_data_science_team-0.0.0.9014/ai_data_science_team.egg-info → ai_data_science_team-0.0.0.9016}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: ai-data-science-team
-Version: 0.0.0.9014
+Version: 0.0.0.9016
 Summary: Build and run an AI-powered data science team.
 Home-page: https://github.com/business-science/ai-data-science-team
 Author: Matt Dancho
@@ -47,6 +47,7 @@ Dynamic: classifier
 Dynamic: description
 Dynamic: description-content-type
 Dynamic: home-page
+Dynamic: license-file
 Dynamic: provides-extra
 Dynamic: requires-dist
 Dynamic: requires-python
@@ -97,9 +98,8 @@ The AI Data Science Team of Copilots includes Agents that specialize data cleani
   - [Companies That Want A Custom AI Data Science Team (And AI Apps)](#companies-that-want-a-custom-ai-data-science-team-and-ai-apps)
   - [Generative AI for Data Scientists Workshop](#generative-ai-for-data-scientists-workshop)
   - [Data Science Agents](#data-science-agents)
+    - [🔥 NEW: Data Science Apps](#-new-data-science-apps)
     - [NEW: Multi-Agents](#new-multi-agents)
-    - [Data Science Apps](#data-science-apps)
-    - [Apps Available Now](#apps-available-now)
       - [🔥 Agentic Applications](#-agentic-applications)
     - [Agents Available Now](#agents-available-now)
       - [Standard Agents](#standard-agents)
@@ -110,11 +110,11 @@ The AI Data Science Team of Copilots includes Agents that specialize data cleani
   - [Disclaimer](#disclaimer)
   - [Installation](#installation)
   - [Usage](#usage)
-    - [Example 1: Feature Engineering with the Feature Engineering Agent](#example-1-feature-engineering-with-the-feature-engineering-agent)
-    - [Example 2: Cleaning Data with the Data Cleaning Agent](#example-2-cleaning-data-with-the-data-cleaning-agent)
+    - [Example: H2O Machine Learning Agent](#example-h2o-machine-learning-agent)
   - [Contributing](#contributing)
   - [License](#license)
 - [Want To Become A Full-Stack Generative AI Data Scientist?](#want-to-become-a-full-stack-generative-ai-data-scientist)
+- [⭐️ Star History](#️-star-history)
 
 ## Companies That Want A Custom AI Data Science Team (And AI Apps)
 
@@ -134,21 +134,24 @@ This project is a work in progress. New data science agents will be released soo
 
 ![AI Data Science Team](/img/ai_data_science_team.jpg) 
 
-### NEW: Multi-Agents
+### 🔥 NEW: Data Science Apps
 
-This is the internals of the SQL Data Analyst Agent that connects to SQL databases to pull data into the data science environment. It creates pipelines to automate data extraction, performs Joins, Aggregations, and other SQL Query operations. And it includes a Data Visualization Agent that creates visualizations to help you understand your data.:
+**🔥 Open Pandas AI Data Analyst:** Load an Excel or CSV file and ask it questions. Get data and charts back. 
 
-![Business Intelligence SQL Agent](/img/multi_agent_sql_data_visualization.jpg)
+![Pandas Data Analyst App](/img/apps/ai_pandas_data_analyst_app.jpg)
 
-### Data Science Apps
+**🔥 SQL Database Agent:** Connects any SQL Database, generates SQL queries from natural language, and returns data as a downloadable table.
 
-This is a top secret project I'm working on. It's a multi-agent data science app that performs time series forecasting.
+**🔥 Exploratory Data Copilot:** An AI-powered data science app that performs automated exploratory data analysis (EDA) with EDA Reporting, Missing Data Analysis, Correlation Analysis, and more.
 
-![Multi-Agent Data Science App](/img/ai_powered_apps.jpg) 
+[See all available apps here](/apps)
 
-### Apps Available Now
+### NEW: Multi-Agents
+
+**🔥 Pandas Data Analyst Agent:** Combines the ability to wrangle, transform, and analyze data with an optional data visualization agent that can create interactive plots.
+
+![Pandas Data Analyst Agent](/img/multi_agent_pandas_data_analyst.jpg)
 
-[See all available apps here](/apps)
 
 #### 🔥 Agentic Applications
 
@@ -182,7 +185,8 @@ This is a top secret project I'm working on. It's a multi-agent data science app
 
 #### Multi-Agents
 
-1. **SQL Data Analyst Agent:** Connects to SQL databases to pull data into the data science environment. Creates pipelines to automate data extraction. Performs Joins, Aggregations, and other SQL Query operations. Includes a Data Visualization Agent that creates visualizations to help you understand your data. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/multiagents/sql_data_analyst.ipynb)
+1. **🔥🔥 Pandas Data Analyst Agent:** Combines the ability to wrangle, transform, and analyze data with an optional data visualization agent that can create interactive plots. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/multiagents/pandas_data_analyst.ipynb)
+2. **🔥🔥 SQL Data Analyst Agent:** Connects to SQL databases to pull data into the data science environment. Creates pipelines to automate data extraction. Performs Joins, Aggregations, and other SQL Query operations. Includes a Data Visualization Agent that creates visualizations to help you understand your data. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/multiagents/sql_data_analyst.ipynb)
 
 ### Agents Coming Soon
 
@@ -204,6 +208,14 @@ By using this software, you agree to use it solely for learning purposes.
 
 ## Installation
 
+You can install via PyPI (note that this is a beta version and breaking changes may occur until 0.1.0):
+
+``` bash
+pip install ai-data-science-team
+```
+
+Or, if you want the latest version from GitHub:
+
 ``` bash
 pip install git+https://github.com/business-science/ai-data-science-team.git --upgrade
 ```
@@ -212,55 +224,46 @@ pip install git+https://github.com/business-science/ai-data-science-team.git --u
 
 [See all examples here.](/examples)
 
-### Example 1: Feature Engineering with the Feature Engineering Agent
+### Example: H2O Machine Learning Agent
 
-[See the full example here.](/examples/feature_engineering_agent.ipynb)
+[See the full example here.](https://github.com/business-science/ai-data-science-team/blob/master/examples/ml_agents/h2o_machine_learning_agent.ipynb)
 
 ``` python
-feature_engineering_agent = FeatureEngineeringAgent(model = llm)
-
-feature_engineering_agent.invoke_agent(
-    data_raw = df,
-    user_instructions = "Make sure to scale and center numeric features",
-    target_variable = "Churn",
-    max_retries = 3,
+# Import libraries
+from langchain_openai import ChatOpenAI
+import pandas as pd
+import h2o 
+import os
+from ai_data_science_team.ml_agents import H2OMLAgent
+
+# Load the data
+df = pd.read_csv("data/churn_data.csv")
+df
+
+# Initialize the language model
+os.environ['OPENAI_API_KEY'] = "YOUR_OPENAI_API_KEY"
+llm = ChatOpenAI(model=MODEL)
+llm
+
+# Initialize the H2O ML Agent
+ml_agent = H2OMLAgent(
+    model=llm, 
+    log=True, 
+    log_path="logs/",
+    model_directory="h2o_models/", 
+    enable_mlflow=True, # Use this if you wish to log models to MLflow 
 )
-```
-
-``` bash
----FEATURE ENGINEERING AGENT----
-    * CREATE FEATURE ENGINEER CODE
-    * EXECUTING AGENT CODE
-    * EXPLAIN AGENT CODE
-```
-
-``` python
-feature_engineering_agent.get_data_engineered()
-```
-
-### Example 2: Cleaning Data with the Data Cleaning Agent
-
-[See the full example here.](/examples/data_cleaning_agent.ipynb) 
-
-``` python
-data_cleaning_agent = DataCleaningAgent(model = llm)
+ml_agent
 
-response = data_cleaning_agent.invoke_agent(
-    data_raw = df,
-    user_instructions = "Don't remove outliers when cleaning the data.",
-    max_retries = 3,
+# Run the agent
+ml_agent.invoke_agent(
+    data_raw=df.drop(columns=["customerID"]),
+    user_instructions="Please do classification on 'Churn'. Use a max runtime of 30 seconds.",
+    target_variable="Churn"
 )
-```
 
-``` bash
----DATA CLEANING AGENT----
-    * CREATE DATA CLEANER CODE
-    * EXECUTING AGENT CODE
-    * EXPLAIN AGENT CODE
-```
-
-``` python
-data_cleaning_agent.get_data_cleaned()
+# Retrieve and display the leaderboard of models
+ml_agent.get_leaderboard()
 ```
 
 ## Contributing
@@ -281,4 +284,8 @@ This project is licensed under the MIT License. See LICENSE file for details.
 
 I teach Generative AI Data Science to help you build AI-powered data science apps. [**Register for my next Generative AI for Data Scientists workshop here.**](https://learn.business-science.io/ai-register)
 
+# ⭐️ Star History
+
+[![Star History Chart](https://api.star-history.com/svg?repos=business-science/ai-data-science-team&type=Date)](https://star-history.com/#)
 
+[**Please ⭐ us on GitHub (it takes 2 seconds and means a lot).**](https://github.com/business-science/ai-data-science-team)

{ai_data_science_team-0.0.0.9014 → ai_data_science_team-0.0.0.9016}/README.md

@@ -43,9 +43,8 @@ The AI Data Science Team of Copilots includes Agents that specialize data cleani
   - [Companies That Want A Custom AI Data Science Team (And AI Apps)](#companies-that-want-a-custom-ai-data-science-team-and-ai-apps)
   - [Generative AI for Data Scientists Workshop](#generative-ai-for-data-scientists-workshop)
   - [Data Science Agents](#data-science-agents)
+    - [🔥 NEW: Data Science Apps](#-new-data-science-apps)
     - [NEW: Multi-Agents](#new-multi-agents)
-    - [Data Science Apps](#data-science-apps)
-    - [Apps Available Now](#apps-available-now)
       - [🔥 Agentic Applications](#-agentic-applications)
     - [Agents Available Now](#agents-available-now)
       - [Standard Agents](#standard-agents)
@@ -56,11 +55,11 @@ The AI Data Science Team of Copilots includes Agents that specialize data cleani
   - [Disclaimer](#disclaimer)
   - [Installation](#installation)
   - [Usage](#usage)
-    - [Example 1: Feature Engineering with the Feature Engineering Agent](#example-1-feature-engineering-with-the-feature-engineering-agent)
-    - [Example 2: Cleaning Data with the Data Cleaning Agent](#example-2-cleaning-data-with-the-data-cleaning-agent)
+    - [Example: H2O Machine Learning Agent](#example-h2o-machine-learning-agent)
   - [Contributing](#contributing)
   - [License](#license)
 - [Want To Become A Full-Stack Generative AI Data Scientist?](#want-to-become-a-full-stack-generative-ai-data-scientist)
+- [⭐️ Star History](#️-star-history)
 
 ## Companies That Want A Custom AI Data Science Team (And AI Apps)
 
@@ -80,21 +79,24 @@ This project is a work in progress. New data science agents will be released soo
 
 ![AI Data Science Team](/img/ai_data_science_team.jpg) 
 
-### NEW: Multi-Agents
+### 🔥 NEW: Data Science Apps
 
-This is the internals of the SQL Data Analyst Agent that connects to SQL databases to pull data into the data science environment. It creates pipelines to automate data extraction, performs Joins, Aggregations, and other SQL Query operations. And it includes a Data Visualization Agent that creates visualizations to help you understand your data.:
+**🔥 Open Pandas AI Data Analyst:** Load an Excel or CSV file and ask it questions. Get data and charts back. 
 
-![Business Intelligence SQL Agent](/img/multi_agent_sql_data_visualization.jpg)
+![Pandas Data Analyst App](/img/apps/ai_pandas_data_analyst_app.jpg)
 
-### Data Science Apps
+**🔥 SQL Database Agent:** Connects any SQL Database, generates SQL queries from natural language, and returns data as a downloadable table.
 
-This is a top secret project I'm working on. It's a multi-agent data science app that performs time series forecasting.
+**🔥 Exploratory Data Copilot:** An AI-powered data science app that performs automated exploratory data analysis (EDA) with EDA Reporting, Missing Data Analysis, Correlation Analysis, and more.
 
-![Multi-Agent Data Science App](/img/ai_powered_apps.jpg) 
+[See all available apps here](/apps)
 
-### Apps Available Now
+### NEW: Multi-Agents
+
+**🔥 Pandas Data Analyst Agent:** Combines the ability to wrangle, transform, and analyze data with an optional data visualization agent that can create interactive plots.
+
+![Pandas Data Analyst Agent](/img/multi_agent_pandas_data_analyst.jpg)
 
-[See all available apps here](/apps)
 
 #### 🔥 Agentic Applications
 
@@ -128,7 +130,8 @@ This is a top secret project I'm working on. It's a multi-agent data science app
 
 #### Multi-Agents
 
-1. **SQL Data Analyst Agent:** Connects to SQL databases to pull data into the data science environment. Creates pipelines to automate data extraction. Performs Joins, Aggregations, and other SQL Query operations. Includes a Data Visualization Agent that creates visualizations to help you understand your data. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/multiagents/sql_data_analyst.ipynb)
+1. **🔥🔥 Pandas Data Analyst Agent:** Combines the ability to wrangle, transform, and analyze data with an optional data visualization agent that can create interactive plots. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/multiagents/pandas_data_analyst.ipynb)
+2. **🔥🔥 SQL Data Analyst Agent:** Connects to SQL databases to pull data into the data science environment. Creates pipelines to automate data extraction. Performs Joins, Aggregations, and other SQL Query operations. Includes a Data Visualization Agent that creates visualizations to help you understand your data. [See Example](https://github.com/business-science/ai-data-science-team/blob/master/examples/multiagents/sql_data_analyst.ipynb)
 
 ### Agents Coming Soon
 
@@ -150,6 +153,14 @@ By using this software, you agree to use it solely for learning purposes.
 
 ## Installation
 
+You can install via PyPI (note that this is a beta version and breaking changes may occur until 0.1.0):
+
+``` bash
+pip install ai-data-science-team
+```
+
+Or, if you want the latest version from GitHub:
+
 ``` bash
 pip install git+https://github.com/business-science/ai-data-science-team.git --upgrade
 ```
@@ -158,55 +169,46 @@ pip install git+https://github.com/business-science/ai-data-science-team.git --u
 
 [See all examples here.](/examples)
 
-### Example 1: Feature Engineering with the Feature Engineering Agent
+### Example: H2O Machine Learning Agent
 
-[See the full example here.](/examples/feature_engineering_agent.ipynb)
+[See the full example here.](https://github.com/business-science/ai-data-science-team/blob/master/examples/ml_agents/h2o_machine_learning_agent.ipynb)
 
 ``` python
-feature_engineering_agent = FeatureEngineeringAgent(model = llm)
-
-feature_engineering_agent.invoke_agent(
-    data_raw = df,
-    user_instructions = "Make sure to scale and center numeric features",
-    target_variable = "Churn",
-    max_retries = 3,
+# Import libraries
+from langchain_openai import ChatOpenAI
+import pandas as pd
+import h2o 
+import os
+from ai_data_science_team.ml_agents import H2OMLAgent
+
+# Load the data
+df = pd.read_csv("data/churn_data.csv")
+df
+
+# Initialize the language model
+os.environ['OPENAI_API_KEY'] = "YOUR_OPENAI_API_KEY"
+llm = ChatOpenAI(model=MODEL)
+llm
+
+# Initialize the H2O ML Agent
+ml_agent = H2OMLAgent(
+    model=llm, 
+    log=True, 
+    log_path="logs/",
+    model_directory="h2o_models/", 
+    enable_mlflow=True, # Use this if you wish to log models to MLflow 
 )
-```
-
-``` bash
----FEATURE ENGINEERING AGENT----
-    * CREATE FEATURE ENGINEER CODE
-    * EXECUTING AGENT CODE
-    * EXPLAIN AGENT CODE
-```
-
-``` python
-feature_engineering_agent.get_data_engineered()
-```
-
-### Example 2: Cleaning Data with the Data Cleaning Agent
-
-[See the full example here.](/examples/data_cleaning_agent.ipynb) 
-
-``` python
-data_cleaning_agent = DataCleaningAgent(model = llm)
+ml_agent
 
-response = data_cleaning_agent.invoke_agent(
-    data_raw = df,
-    user_instructions = "Don't remove outliers when cleaning the data.",
-    max_retries = 3,
+# Run the agent
+ml_agent.invoke_agent(
+    data_raw=df.drop(columns=["customerID"]),
+    user_instructions="Please do classification on 'Churn'. Use a max runtime of 30 seconds.",
+    target_variable="Churn"
 )
-```
 
-``` bash
----DATA CLEANING AGENT----
-    * CREATE DATA CLEANER CODE
-    * EXECUTING AGENT CODE
-    * EXPLAIN AGENT CODE
-```
-
-``` python
-data_cleaning_agent.get_data_cleaned()
+# Retrieve and display the leaderboard of models
+ml_agent.get_leaderboard()
 ```
 
 ## Contributing
@@ -227,4 +229,8 @@ This project is licensed under the MIT License. See LICENSE file for details.
 
 I teach Generative AI Data Science to help you build AI-powered data science apps. [**Register for my next Generative AI for Data Scientists workshop here.**](https://learn.business-science.io/ai-register)
 
+# ⭐️ Star History
+
+[![Star History Chart](https://api.star-history.com/svg?repos=business-science/ai-data-science-team&type=Date)](https://star-history.com/#)
 
+[**Please ⭐ us on GitHub (it takes 2 seconds and means a lot).**](https://github.com/business-science/ai-data-science-team)

ai_data_science_team-0.0.0.9016/ai_data_science_team/_version.py

@@ -0,0 +1 @@
+__version__ = "0.0.0.9016"