L
Initializing Studio...
Discover the powerful features that make LangTrain the best platform for training and deploying language models.
1import langtrain23client = langtrain.LangTrain()45# Create model with auto-tuning6model = client.models.create(7 name="optimized-classifier",8 type="text-classification"9)1011# Train with auto-tuning enabled12training_job = model.train(13 dataset_id="your-dataset-id",14 auto_tune=True,15 auto_tune_config={16 "max_trials": 50,17 "optimization_metric": "f1_score",18 "search_space": {19 "learning_rate": [1e-5, 5e-4],20 "batch_size": [16, 32, 64],21 "num_epochs": [3, 5, 10]22 }23 }24)2526# Get best configuration27best_config = training_job.get_best_config()28print(f"Best learning rate: {best_config['learning_rate']}")29print(f"Best batch size: {best_config['batch_size']}")