Commit 4cde14df authored by okiss's avatar okiss
Browse files

Update DeepEye.py

parent a26144b2
......@@ -17,6 +17,20 @@ import matplotlib.pyplot as plt
import seaborn as sns
sns.set_style('darkgrid')
import os
from sklearn.model_selection import train_test_split
def run(trainX,trainY):
nb_class=2
input_shape=np.shape(trainX)
output_dir=os.getcwd()
x_train, x_val, y_train, y_val = train_test_split(trainX, trainY, test_size=0.2,shuffle=True, random_state=42)
deepeye_classifier=Classifier_DEEPEYE(output_directory, input_shape, nb_classes)
df_metrics=deepeye_classifier.fit(x_train, y_train, x_val, y_val, y_true, plot_test_acc=True)
print(10*'*',df_metrics)
class Classifier_DEEPEYE:
"""
Inputs:
......@@ -285,3 +299,7 @@ class Classifier_DEEPEYE:
plt.xlabel('epochs')
plt.ylabel('Binary Cross Entropie')
plt.savefig(self.output_directory+'loss_'+name+'.png')
plt.legend()
plt.xlabel('epochs')
plt.ylabel('Binary Cross Entropie')
plt.savefig(self.output_directory+'loss_'+name+'.png')
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment