pip install augini
from augini import Augini
import pandas as pd
api_key = "OpenAI or OpenRouter"
# OpenAI
augini = Augini(api_key=api_key, model='gpt-4-turbo', use_openrouter=False)
# OpenRouter
augini = Augini(api_key=api_key, use_openrouter=True, model='meta-llama/llama-3-8b-instruct')
# Create a sample DataFrame
data = {
'Place of Birth': ['New York', 'London', 'Tokyo'],
'Age': [30, 25, 40],
'Gender': ['Male', 'Female', 'Male']
}
df = pd.DataFrame(data)
# Add synthetic features
result_df = augini.augment_columns(df, ['NAME', 'OCCUPATION', 'FAVORITE_DRINK'])
print(result_df)