-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathqc.backup.py
74 lines (61 loc) · 2.57 KB
/
qc.backup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
from Params import *
from load_data import *
from preprocess import *
from vgg_like_convnet import *
from alexnet import *
from vgg16_keras import *
from pretrained import *
import h5py
import keras
import theano
#for visualization:
from keras.utils.visualize_util import plot
from keras.optimizers import SGD, Adagrad, Adadelta
import pickle
import numpy as np
import sys
def main():
#load data
#X_train,Y_train,X_valid,Y_valid,X_test=load_data(training_dir,valid_dir,test_dir,labels,sample)
#preprocess data by mean subtraction and normalization
#X_train,X_valid,X_test=preprocess(X_train,X_valid,X_test)
#del X_train
#del X_test
#or load pre-processed data from a previously saved hdf5 file:
data=h5py.File('imagenet.transpose.individually.hdf5','r')
X_train=np.asarray(data['X_train'])
Y_train=np.asarray(data['Y_train'])
X_valid=np.asarray(data['X_valid'])
Y_valid=np.asarray(data['Y_valid'])
X_test=np.asarray(data['X_test'])
pretrained_model=pretrained_finetune('assignment3_weights_nodropout_noregularization_augmenteddata.hdf5',freezeAndStack=False)
sgd = SGD(lr=1e-1)#, decay=1e-6, momentum=0.9, nesterov=True)
pretrained_model.compile(optimizer=sgd, loss='categorical_crossentropy',trainLayersIndividually=0)
valid_scores=pretrained_evaluate(pretrained_model,X_valid,Y_valid)
print "pretrained training scores:"+str(valid_scores)
#Visualize the pretty model
#plot(pretrained_model,to_file="pretrained_convnet.png")
#run the model on our test data
print "predicting on evaluation data:"
#evaluate_predictions=pretrained_model.predict(X_valid,verbose=1)
#np.savetxt('valid.out', evaluate_predictions, fmt='%f')
#print "predicting on test data:"
#predictions=pretrained_model.predict(X_test,verbose=1)
#np.savetxt('test.raw.out',predictions,fmt='%f')
#print "getting class predictions on test data:"
#class_predictions=pretrained_model.predict_classes(X_train)
#np.savetxt('train.out',class_predictions,fmt='%i')
#print "validation data predictions:"+str(evaluate_predictions)
#print "test predictions:"+str(predictions)
#save all the outputs!
#sys.setrecursionlimit(50000)
#outputf=open('qc.pkl','w')
#output=open('pretrained_results_freezeAndStack.pkl','wb')
#pickle.dump(history,outputf)
#pickle.dump(train_scores,outputf)
#pickle.dump(valid_scores,outputf)
#pickle.dump(predictions,outputf)
#pickle.dump(class_predictions,outputf)
#outputf.close()
if __name__=="__main__":
main()