fit

The fit method learns to generate enhanced statistical properties from the inputs and target variables using Ensemble's proprietary Feature Enhancement algorithm.

fit(X: pandas.DataFrame|numpy.ndarray|torch.tensor,

y: pandas.DataFrame|numpy.ndarray|torch.tensor) -> None

Example:

import ensemble_core as ec

# Login
user = ec.User()
user.login(username="YOUR_USERNAME", password="YOUR_PASSWORD", token="YOUR_TOKEN")

# Example data
X = np.array([[1, 1], [1, 2], [2, 2], [2, 3]])
y = np.dot(X, np.array([1, 2])) + 3

# Initialize generator with your desired specifications
generator = ec.Generator(in_size=2, out_size=1, n_features=5)

# Run Feature Enhancement
generator.fit(X, y)           

# Generate new and enhanced features
enhanced_X = generator.generate(X)  

Last updated