diff --git a/src/models/prediction.ts b/src/models/prediction.ts new file mode 100644 index 0000000..b379eaf --- /dev/null +++ b/src/models/prediction.ts @@ -0,0 +1,8 @@ +import mongoose from 'mongoose'; + + +const PredictionSchema = new mongoose.Schema({ + values: [{ spectrumName: String, value: Number }] +}); + +export default mongoose.model>('prediction', PredictionSchema); diff --git a/src/models/user.ts b/src/models/user.ts index a70791b..51ad234 100644 --- a/src/models/user.ts +++ b/src/models/user.ts @@ -1,6 +1,7 @@ import mongoose from 'mongoose'; import db from '../db'; import ModelModel from './model'; +import PredictionModel from './prediction'; const UserSchema = new mongoose.Schema({ name: {type: String, index: {unique: true}}, @@ -11,6 +12,7 @@ const UserSchema = new mongoose.Schema({ location: String, devices: [String], models: [{type: mongoose.Schema.Types.ObjectId, ref: ModelModel}], + predictions: [{type: mongoose.Schema.Types.ObjectId, ref: PredictionModel}], status: String });