import plexe
import pandas as pd
# Load your data into DataFrames
try:
train_df = pd.read_csv("train_data.csv")
# val_df = pd.read_csv("validation_data.csv") # Optional validation set
except FileNotFoundError:
print("Warning: Data files not found, using dummy data.")
train_df = pd.DataFrame({
'feature1': [1, 2, 3, 4, 5], 'feature2': [10, 12, 11, 14, 13], 'target': [0, 1, 0, 1, 0]
})
# val_df = pd.DataFrame({ # Dummy validation data
# 'feature1': [6, 7], 'feature2': [15, 16], 'target': [1, 0]
# })
# --- Define Model ---
model = plexe.Model(
intent="Classify target based on feature1 and feature2."
# Schemas might be inferred if not provided explicitly
)
# --------------------
# Provide the DataFrame(s) in a list
datasets_to_use = [train_df]
# datasets_to_use = [train_df, val_df] # If you have a separate validation set
print("Building model using Pandas DataFrame(s)...")
model.build(
datasets=datasets_to_use,
provider="openai/gpt-4o-mini",
max_iterations=1
)
print(f"Model build finished. State: {model.get_state()}")