Skip to content

Instantly share code, notes, and snippets.

@BinarySpoon

BinarySpoon/Fitting model.py Secret

Last active October 8, 2020 00:02
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save BinarySpoon/7521d3187b7e5253fc5e296ecb8f3bae to your computer and use it in GitHub Desktop.
Save BinarySpoon/7521d3187b7e5253fc5e296ecb8f3bae to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Fitting model.py
# Training Classifier
#clf = SVC(kernel = 'rbf', gamma=0.01, C=0.1)
#clf = LogisticRegression()
#clf = KneighborsRegression(n_neighbours = 5)
clf = MLPClassifier(hidden_layer_sizes=[100,10],alpha=0.001,random_state=0,solver='lbfgs',verbose=0)
clf.fit(X_train_scaled, y_train)
# Predict probabilities
y_proba = clf.predict_proba(X_test_scaled)[:,1]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment