<commit_before>import java.net.UnknownHostException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Map;
import java.util.TreeMap;

import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBCursor;
import com.mongodb.DBObject;
import com.mongodb.MongoClient;


public class MongoWrapper implements IWrapper{

	private String collectionName;
	private ArrayList<String> parameters;
	private MongoClient mc;
	private DB db;
	private DBCollection coll;

	public static void main(String[]arg) throws UnknownHostException{
		MongoWrapper mw = new MongoWrapper();
		mw.trainNetwork();
	}
	
	public MongoWrapper() throws UnknownHostException{
		collectionName = "FinalOutput";
		parameters = new ArrayList<String>();
		//Populate list here using input string

		mc = new MongoClient();
		db = mc.getDB("MNISTDB");
		coll = db.getCollection("test");
	}

	public String trainNetwork(){
		
		DBCursor curs = coll.find();
		DBObject o = curs.next();
		
		Double a = (Double)o.get("test");
		System.out.println(a);
		Double b = (Double)o.get("label");
		System.out.println(b);
		Double[][] c = (Double[][]) o.get("image");
		
		return null;
		// RUN PREDICTION HERE
	}

	public int predict(){
		return 0;
	}

}
<commit_msg>wefaeafs<commit_after>import java.net.UnknownHostException;
import java.util.ArrayList;
import com.mongodb.DB;
import com.mongodb.DBCollection;
import com.mongodb.DBCursor;
import com.mongodb.DBObject;
import com.mongodb.MongoClient;


public class MongoWrapper implements IWrapper{

	private String collectionName;
	private ArrayList<String> parameters;
	private MongoClient mc;
	private DB db;
	private DBCollection coll;

	public static void main(String[]arg) throws UnknownHostException{
		MongoWrapper mw = new MongoWrapper();
		mw.trainNetwork();
	}
	
	public MongoWrapper() throws UnknownHostException{
		collectionName = "FinalOutput";
		parameters = new ArrayList<String>();
		//Populate list here using input string

		mc = new MongoClient();
		db = mc.getDB("MNISTDB");
		coll = db.getCollection("test");
	}

	public String trainNetwork(){
		
		DBCursor curs = coll.find();
		DBObject o = curs.next();
		
		Double a = (Double)o.get("test");
		System.out.println(a);
		Double b = (Double)o.get("label");
		System.out.println(b);
		Double[][] c = (Double[][]) o.get("image");
		
		return null;
		// RUN PREDICTION HERE
	}

	public int predict(){
		return 0;
	}

}

{'fraction_non_alphanumeric': 0.12551440329218108, 'fraction_numerical': 0.0008230452674897119, 'mean_word_length': 4.525, 'pattern_counts': {'":': 0, '<': 7, '<?xml version=': 0, '>': 7, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '2933390', 'n_tokens_mistral': 863, 'n_tokens_neox': 758, 'n_words': 202}