Skip to content

Saroj Dhiman / nlp_to_sql_query_generator

Go to a project
Commit 06c8515d by Saroj Dhiman
Options

Update vanna_model2.py

1 parent 6c136aeb
Inline Side-by-side
Showing with 7 additions and 5 deletions
vanna_model2.py
......@@ -8,8 +8,8 @@ vn.set_api_key(api_key)
# Create models
vn.create_model(model="data_warehose", db_type="Snowflake")
vn.create_model(model="marketing_db", db_type="Postgres")
vn.create_model(model="marketing_db", db_type="BigQuery")
# vn.create_model(model="marketing_db", db_type="Postgres")
# vn.create_model(model="marketing_db", db_type="BigQuery")
models = vn.get_models()
......@@ -17,15 +17,17 @@ models = vn.get_models()
vn.set_model('chinook')
# Connect to database
url= "database.sqlite3"
vn.connect_to_sqlite(url=url)
url= "https://git.ociustechnologies.com/saroj.dhiman/nlp_to_sql_query_generator/raw/vanna_branch/database.sqlite3"
data1 = vn.connect_to_sqlite(url=url)
print(data1)
# Check training data
training_data = vn.get_training_data()
training_data
vn.ask(question='What are the most occured cancer in CancerData?')
vn.ask(question='What are the cancer in CancerData?')
# training_data = vn.get_training_data()
# training_data['question'].str.contains('total Stomach records in cancertable?').any()
......
Markdown is supported
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!