ai-data-science-team 0.0.0.9000__py3-none-any.whl → 0.0.0.9005__py3-none-any.whl
Sign up to get free protection for your applications and to get access to all the features.
- ai_data_science_team/_version.py +1 -1
- ai_data_science_team/agents/__init__.py +4 -0
- ai_data_science_team/agents/data_cleaning_agent.py +347 -0
- ai_data_science_team/agents/data_wrangling_agent.py +365 -0
- ai_data_science_team/agents/feature_engineering_agent.py +368 -0
- ai_data_science_team/templates/__init__.py +0 -0
- ai_data_science_team/templates/agent_templates.py +409 -0
- ai_data_science_team/tools/__init__.py +0 -0
- ai_data_science_team/tools/data_analysis.py +116 -0
- ai_data_science_team/tools/logging.py +61 -0
- ai_data_science_team/tools/parsers.py +57 -0
- ai_data_science_team/tools/regex.py +73 -0
- ai_data_science_team-0.0.0.9005.dist-info/METADATA +162 -0
- ai_data_science_team-0.0.0.9005.dist-info/RECORD +19 -0
- ai_data_science_team/agents.py +0 -325
- ai_data_science_team-0.0.0.9000.dist-info/METADATA +0 -131
- ai_data_science_team-0.0.0.9000.dist-info/RECORD +0 -9
- {ai_data_science_team-0.0.0.9000.dist-info → ai_data_science_team-0.0.0.9005.dist-info}/LICENSE +0 -0
- {ai_data_science_team-0.0.0.9000.dist-info → ai_data_science_team-0.0.0.9005.dist-info}/WHEEL +0 -0
- {ai_data_science_team-0.0.0.9000.dist-info → ai_data_science_team-0.0.0.9005.dist-info}/top_level.txt +0 -0
@@ -1,131 +0,0 @@
|
|
1
|
-
Metadata-Version: 2.1
|
2
|
-
Name: ai-data-science-team
|
3
|
-
Version: 0.0.0.9000
|
4
|
-
Summary: Build and run an AI-powered data science team.
|
5
|
-
Home-page: https://github.com/business-science/ai-data-science-team
|
6
|
-
Author: Matt Dancho
|
7
|
-
Author-email: mdancho@business-science.io
|
8
|
-
Requires-Python: >=3.9
|
9
|
-
Description-Content-Type: text/markdown
|
10
|
-
License-File: LICENSE
|
11
|
-
Requires-Dist: openpyxl
|
12
|
-
Requires-Dist: langchain
|
13
|
-
Requires-Dist: langchain_community
|
14
|
-
Requires-Dist: langchain_openai
|
15
|
-
Requires-Dist: langchain_experimental
|
16
|
-
Requires-Dist: langgraph
|
17
|
-
Requires-Dist: openai
|
18
|
-
Requires-Dist: pandas
|
19
|
-
Requires-Dist: numpy
|
20
|
-
Requires-Dist: plotly
|
21
|
-
Requires-Dist: streamlit
|
22
|
-
Requires-Dist: scikit-learn
|
23
|
-
Requires-Dist: xgboost
|
24
|
-
|
25
|
-
# AI Data Science Team
|
26
|
-
|
27
|
-
**An AI-powered data science team that uses agents to perform common data science** tasks including data cleaning, preparation, feature engineering, modeling (machine learning), interpretation on various business problems like:
|
28
|
-
|
29
|
-
- Churn Modeling
|
30
|
-
- Employee Attrition
|
31
|
-
- Lead Scoring
|
32
|
-
- Insurance Risk
|
33
|
-
- Credit Card Risk
|
34
|
-
- And more
|
35
|
-
|
36
|
-
## Companies That Want An AI Data Science Team
|
37
|
-
|
38
|
-
If you are interested in having your own AI Data Science Team built and deployed for your enterprise, send inquiries here: [https://www.business-science.io/contact.html](https://www.business-science.io/contact.html)
|
39
|
-
|
40
|
-
## Free Generative AI For Data Scientists Workshop
|
41
|
-
|
42
|
-
If you want to learn how to build AI Agents for your company that perform Data Science, Business Intelligence, Churn Modeling, Time Series Forecasting, and more, [register for my next Generative AI for Data Scientists workshop here.](https://learn.business-science.io/ai-register)
|
43
|
-
|
44
|
-
## Agents
|
45
|
-
|
46
|
-
This project is a work in progress. New agents will be released soon.
|
47
|
-
|
48
|
-

|
49
|
-
|
50
|
-
### Agents Available Now:
|
51
|
-
|
52
|
-
1. **Data Cleaning Agent:** Performs Data Preparation steps including handling missing values, outliers, and data type conversions.
|
53
|
-
|
54
|
-
### Agents Coming Soon:
|
55
|
-
|
56
|
-
1. **Supervisor:** Forms task list. Moderates sub-agents. Returns completed assignment.
|
57
|
-
2. **Exploratory Data Analyst:** Analyzes data structure, creates exploratory visualizations, and performs correlation analysis to identify relationships.
|
58
|
-
3. **Feature Engineering Agent:** Converts the prepared data into ML-ready data. Adds features to increase predictive accuracy of ML models.
|
59
|
-
4. **Machine Learning Agent:** Builds and logs the machine learning models.
|
60
|
-
5. **Interpretability Agent:** Performs Interpretable ML to explain why the model returned predictions including which features were the most important to the model.
|
61
|
-
|
62
|
-
## Disclaimer
|
63
|
-
|
64
|
-
**This project is for educational purposes only.**
|
65
|
-
|
66
|
-
- It is not intended to replace your company's data science team
|
67
|
-
- No warranties or guarantees provided
|
68
|
-
- Creator assumes no liability for financial loss
|
69
|
-
- Consult an experienced Generative AI Data Scientist for building your own AI Data Science Team
|
70
|
-
- If you want an enterprise-grade AI Data Science Team, [send inquiries here](https://www.business-science.io/contact.html).
|
71
|
-
|
72
|
-
By using this software, you agree to use it solely for learning purposes.
|
73
|
-
|
74
|
-
## Table of Contents
|
75
|
-
|
76
|
-
- [AI Data Science Team](#ai-data-science-team)
|
77
|
-
- [Companies That Want An AI Data Science Team](#companies-that-want-an-ai-data-science-team)
|
78
|
-
- [Free Generative AI For Data Scientists Workshop](#free-generative-ai-for-data-scientists-workshop)
|
79
|
-
- [Agents](#agents)
|
80
|
-
- [Agents Available Now:](#agents-available-now)
|
81
|
-
- [Agents Coming Soon:](#agents-coming-soon)
|
82
|
-
- [Disclaimer](#disclaimer)
|
83
|
-
- [Table of Contents](#table-of-contents)
|
84
|
-
- [Installation](#installation)
|
85
|
-
- [Usage](#usage)
|
86
|
-
- [Example 1: Cleaning Data with the Data Cleaning Agent](#example-1-cleaning-data-with-the-data-cleaning-agent)
|
87
|
-
- [Contributing](#contributing)
|
88
|
-
- [License](#license)
|
89
|
-
|
90
|
-
## Installation
|
91
|
-
|
92
|
-
``` bash
|
93
|
-
pip install git+https://github.com/business-science/ai-data-science-team.git --upgrade
|
94
|
-
```
|
95
|
-
|
96
|
-
## Usage
|
97
|
-
|
98
|
-
### Example 1: Cleaning Data with the Data Cleaning Agent
|
99
|
-
|
100
|
-
[See the full example here.](https://github.com/business-science/ai-data-science-team/blob/master/examples/data_cleaning_agent.ipynb)
|
101
|
-
|
102
|
-
``` python
|
103
|
-
data_cleaning_agent = data_cleaning_agent(model = llm, log=LOG, log_path=LOG_PATH)
|
104
|
-
|
105
|
-
response = data_cleaning_agent.invoke({
|
106
|
-
"user_instructions": "Don't remove outliers when cleaning the data.",
|
107
|
-
"data_raw": df.to_dict(),
|
108
|
-
"max_retries":3,
|
109
|
-
"retry_count":0
|
110
|
-
})
|
111
|
-
```
|
112
|
-
|
113
|
-
``` bash
|
114
|
-
---DATA CLEANING AGENT----
|
115
|
-
* CREATE DATA CLEANER CODE
|
116
|
-
* EXECUTING AGENT CODE
|
117
|
-
* EXPLAIN AGENT CODE
|
118
|
-
```
|
119
|
-
|
120
|
-
## Contributing
|
121
|
-
|
122
|
-
1. Fork the repository
|
123
|
-
2. Create a feature branch
|
124
|
-
3. Commit your changes
|
125
|
-
4. Push to the branch
|
126
|
-
5. Create a Pull Request
|
127
|
-
|
128
|
-
## License
|
129
|
-
|
130
|
-
This project is licensed under the MIT License. See LICENSE file for details.
|
131
|
-
|
@@ -1,9 +0,0 @@
|
|
1
|
-
ai_data_science_team/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
2
|
-
ai_data_science_team/_version.py,sha256=804c98kcVfACZVVhiq8Z8G5icGH_Swif7_kFpUgDlI8,26
|
3
|
-
ai_data_science_team/agents.py,sha256=eYc2CFTZ9o4AKZwp_LmLRcm1PggMuKiaQijb2s93fs4,11922
|
4
|
-
ai_data_science_team/orchestration.py,sha256=xiIFOsrLwPdkSmtme7wNCCGv8XopnMTNElNzlZokL-4,303
|
5
|
-
ai_data_science_team-0.0.0.9000.dist-info/LICENSE,sha256=Xif0IRLdd2HGLATxV2EVp91aSY6KOuacRr_6BorKGzA,1084
|
6
|
-
ai_data_science_team-0.0.0.9000.dist-info/METADATA,sha256=KQ-pwCfosLP-vHRBs2AJ8LqHaRi1OsjQ82sHDdnc7Rk,4770
|
7
|
-
ai_data_science_team-0.0.0.9000.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
|
8
|
-
ai_data_science_team-0.0.0.9000.dist-info/top_level.txt,sha256=CnoMgOphCoAdGTLueWdCVByVyjwOubaGiTB1lchdy4M,21
|
9
|
-
ai_data_science_team-0.0.0.9000.dist-info/RECORD,,
|
{ai_data_science_team-0.0.0.9000.dist-info → ai_data_science_team-0.0.0.9005.dist-info}/LICENSE
RENAMED
File without changes
|
{ai_data_science_team-0.0.0.9000.dist-info → ai_data_science_team-0.0.0.9005.dist-info}/WHEEL
RENAMED
File without changes
|
File without changes
|