Machine Learning Engineer
Building AI Solutions

Aspiring AI Engineer/ Eunterpuner passionate about transforming complex data into intelligent systems. Currently building skills in Computer Vision, Natural Language Processing, and Deep Learning as a first-year B.Tech student.

neural_network.py
import tensorflow as tf

class DeepLearningModel:
    def __init__(self, layers):
        self.model = tf.keras.Sequential([
            tf.keras.layers.Dense(128, activation='relu'),
            tf.keras.layers.Dropout(0.3),
            tf.keras.layers.Dense(64, activation='relu'),
            tf.keras.layers.Dense(10, activation='softmax')
        ])
    
    def train(self, X, y):
        self.model.compile(
            optimizer='adam',
            loss='categorical_crossentropy',
            metrics=['accuracy']
        )
        return self.model.fit(X, y, epochs=100)