Skip to content

Commit

Permalink
Merge pull request #58 from ambroggi/dev
Browse files Browse the repository at this point in the history 
Fixed current hyperparams list
  • Loading branch information
@bayegaspard
bayegaspard authored Oct 18, 2023
2 parents 4f6146e + 64d2452 commit 8fbad77
Show file tree
Hide file tree
Showing 4 changed files with 37 additions and 37 deletions.
2 changes: 1 addition & 1 deletion build_number.txt
View file
Original file line number Diff line number Diff line change
@@ -1 +1 @@
510
511
62 changes: 31 additions & 31 deletions datasets/hyperparamList.csv
View file
Original file line number Diff line number Diff line change
@@ -1,31 +1,31 @@
,batch_size,num_workers,testlength,Mix unknowns and validation,MaxPerClass,num_epochs,learningRate,threshold,model,OOD Type,Dropout,Datagrouping,optimizer,Unknowns,CLASSES,Temperature,Degree of Overcompleteness,Number of Layers,Nodes,Activation,Dataset,SchedulerStepSize,SchedulerStep,Unknowns_clss,Knowns_clss
0,200,0,0.25,1,10,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[1],"[0, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[1],"[0, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[1],"[0, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[2],"[0, 1, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[2],"[0, 1, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[2],"[0, 1, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[3],"[0, 1, 2, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[3],"[0, 1, 2, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[3],"[0, 1, 2, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[4],"[0, 1, 2, 3, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[4],"[0, 1, 2, 3, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[4],"[0, 1, 2, 3, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[5],"[0, 1, 2, 3, 4, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[5],"[0, 1, 2, 3, 4, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[5],"[0, 1, 2, 3, 4, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[6],"[0, 1, 2, 3, 4, 5, 7, 8, 9, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[6],"[0, 1, 2, 3, 4, 5, 7, 8, 9, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[6],"[0, 1, 2, 3, 4, 5, 7, 8, 9, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[7],"[0, 1, 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[7],"[0, 1, 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[7],"[0, 1, 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[8],"[0, 1, 2, 3, 4, 5, 6, 7, 9, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[8],"[0, 1, 2, 3, 4, 5, 6, 7, 9, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[8],"[0, 1, 2, 3, 4, 5, 6, 7, 9, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[9],"[0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[9],"[0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[9],"[0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[10],"[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[10],"[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,0.5,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,3 Unknowns,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[10],"[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 12, 13, 14]"
,batch_size,num_workers,testlength,Mix unknowns and validation,MaxPerClass,num_epochs,learningRate,threshold,model,OOD Type,Dropout,Datagrouping,optimizer,CLASSES,Temperature,Degree of Overcompleteness,Number of Layers,Nodes,Activation,Dataset,SchedulerStepSize,SchedulerStep,Unknowns_clss,Knowns_clss
0,200,0,0.25,1,10,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[1],"[0, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[1],"[0, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[1],"[0, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[2],"[0, 1, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[2],"[0, 1, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[2],"[0, 1, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[3],"[0, 1, 2, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[3],"[0, 1, 2, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[3],"[0, 1, 2, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[4],"[0, 1, 2, 3, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[4],"[0, 1, 2, 3, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[4],"[0, 1, 2, 3, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[5],"[0, 1, 2, 3, 4, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[5],"[0, 1, 2, 3, 4, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[5],"[0, 1, 2, 3, 4, 6, 7, 8, 9, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[6],"[0, 1, 2, 3, 4, 5, 7, 8, 9, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[6],"[0, 1, 2, 3, 4, 5, 7, 8, 9, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[6],"[0, 1, 2, 3, 4, 5, 7, 8, 9, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[7],"[0, 1, 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[7],"[0, 1, 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[7],"[0, 1, 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[8],"[0, 1, 2, 3, 4, 5, 6, 7, 9, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[8],"[0, 1, 2, 3, 4, 5, 6, 7, 9, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[8],"[0, 1, 2, 3, 4, 5, 6, 7, 9, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[9],"[0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[9],"[0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[9],"[0, 1, 2, 3, 4, 5, 6, 7, 8, 10, 11, 12, 13, 14]"
0,200,0,0.25,1,10,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[10],"[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 12, 13, 14]"
1,200,0,0.25,1,50,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[10],"[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 12, 13, 14]"
2,200,0,0.25,1,100,10,0.01,3,Convolutional,Var,0.01,Dendrogramlimit,<class 'torch.optim.adam.Adam'>,15,1,3,2,512,Leaky,Payload_data_CICIDS2017,10,0.8,[10],"[0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 11, 12, 13, 14]"
4 changes: 2 additions & 2 deletions src/main/EndLayer.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -330,11 +330,11 @@ def iiUnknown(self, percentages):

return torch.cat([percentages,unknowns.unsqueeze(dim=-1)],dim=-1)

def varmax_final(self, logits):
def varmax_final(self, logits:torch.Tensor):
import CodeFromImplementations.Varmax
varmax_mask = torch.var(torch.abs(logits), dim = 1) > self.cutoff
shape = logits.shape
unknown = torch.zeros([shape[0],1])
unknown = torch.zeros([shape[0],1], device=logits.device)
unknown[varmax_mask] = 2
output = torch.concat([torch.softmax(logits, dim=-1), unknown], dim = -1)
return output
Expand Down
6 changes: 3 additions & 3 deletions src/main/FileHandling.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -500,7 +500,7 @@ def start(self):

class items_with_classes_record():
def __init__(self, labels:torch.Tensor):
self.labels = labels.unsqueeze(dim=-1)
self.labels = labels.unsqueeze(dim=-1).cpu()
self.items = None
self.predict = None

Expand All @@ -509,7 +509,7 @@ def __call__(self, items:torch.Tensor, file = "Saves/items.csv"):
self.useItems(file)

def storeItems(self, items:torch.Tensor):
self.items = items
self.items = items.cpu()

def useItems(self, file = "Saves/items.csv"):
index_names = [f"Logit{x}" for x in range(len(self.items[0]))]
Expand All @@ -526,4 +526,4 @@ def useItems(self, file = "Saves/items.csv"):

def storePredictions(self, predictions:torch.Tensor):
assert predictions.dim() == 1
self.predict = predictions.unsqueeze(dim=-1)
self.predict = predictions.unsqueeze(dim=-1).cpu()

0 comments on commit 8fbad77

Please sign in to comment.